linhcuem commited on
Commit
85b23b5
1 Parent(s): 6ed6046

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def yolov8_img_inference(
26
  conf_threshold: gr.inputs.Slider = 0.25,
27
  iou_threshold: gr.inputs.Slider = 0.45,
28
  ):
29
- model = YOLO(model)
30
  model.overrides['conf'] = conf_threshold
31
  model.overrides['iou']= iou_threshold
32
  model.overrides['agnostic_nms'] = False # NMS class-agnostic
 
26
  conf_threshold: gr.inputs.Slider = 0.25,
27
  iou_threshold: gr.inputs.Slider = 0.45,
28
  ):
29
+ model = YOLO(model_path)
30
  model.overrides['conf'] = conf_threshold
31
  model.overrides['iou']= iou_threshold
32
  model.overrides['agnostic_nms'] = False # NMS class-agnostic