Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ def yolov8_img_inference(
|
|
45 |
# model.overrides['agnostic_nms'] = False
|
46 |
# model.overrides['max_det'] = 1000
|
47 |
# image = read_image
|
48 |
-
results = model.predict(image)
|
49 |
render = render_result(model=model, image=image, result=results[0])
|
50 |
# get the model names list
|
51 |
names = model.names
|
|
|
45 |
# model.overrides['agnostic_nms'] = False
|
46 |
# model.overrides['max_det'] = 1000
|
47 |
# image = read_image
|
48 |
+
results = model.predict(image, imgsz=image_size, conf=conf_threshold, iou=iou_threshold)
|
49 |
render = render_result(model=model, image=image, result=results[0])
|
50 |
# get the model names list
|
51 |
names = model.names
|