vumichien commited on
Commit
3018237
·
1 Parent(s): 34c1ae3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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