linhcuem commited on
Commit
93f44e4
1 Parent(s): ae895c3

Update app.py

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