Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ def yolov8_inference(
|
|
29 |
model.overrides['iou'] = iou_threshold # NMS IoU threshold
|
30 |
model.overrides['agnostic_nms'] = False # NMS class-agnostic
|
31 |
model.overrides['max_det'] = 1000 # maximum number of detections per image
|
32 |
-
results = model.predict(image)
|
33 |
render = render_result(model=model, image=image, result=results[0])
|
34 |
return render
|
35 |
|
|
|
29 |
model.overrides['iou'] = iou_threshold # NMS IoU threshold
|
30 |
model.overrides['agnostic_nms'] = False # NMS class-agnostic
|
31 |
model.overrides['max_det'] = 1000 # maximum number of detections per image
|
32 |
+
results = model.predict(image, imgsz=image_size)
|
33 |
render = render_result(model=model, image=image, result=results[0])
|
34 |
return render
|
35 |
|