Spaces:

linhcuem
/

cham_diem_yolov8

Sleeping

linhcuem commited on Sep 6, 2023

Commit

85b23b5

•

1 Parent(s): 6ed6046

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ def yolov8_img_inference(
     conf_threshold: gr.inputs.Slider = 0.25,
     iou_threshold: gr.inputs.Slider = 0.45,
 ):
-    model = YOLO(model)
     model.overrides['conf'] = conf_threshold
     model.overrides['iou']= iou_threshold
     model.overrides['agnostic_nms'] = False  # NMS class-agnostic

     conf_threshold: gr.inputs.Slider = 0.25,
     iou_threshold: gr.inputs.Slider = 0.45,
 ):
+    model = YOLO(model_path)
     model.overrides['conf'] = conf_threshold
     model.overrides['iou']= iou_threshold
     model.overrides['agnostic_nms'] = False  # NMS class-agnostic