linhcuem commited on
Commit
8b488a0
·
verified ·
1 Parent(s): 345e28e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -48,9 +48,9 @@ def yolov8_img_inference(
48
  # model = YOLO(model_path)
49
  global current_model_id
50
  global model
51
- if model_id != current_model_id:
52
- model = YOLO(model_id)
53
- current_model_id = model_id
54
  model.conf = conf_threshold
55
  model.iou = iou_threshold
56
  # model.overrides['conf'] = conf_threshold
 
48
  # model = YOLO(model_path)
49
  global current_model_id
50
  global model
51
+ if model_path != current_model_id:
52
+ model = YOLO(model_path)
53
+ current_model_id = model_path
54
  model.conf = conf_threshold
55
  model.iou = iou_threshold
56
  # model.overrides['conf'] = conf_threshold