linhcuem commited on
Commit
a608033
·
1 Parent(s): 53eb390

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def predict(image, threshold=0.25, model_id=None):
23
  global current_model_id
24
  global model
25
  if model_id != current_model_id:
26
- model = yolov5.load(models_id)
27
  # model_yolov8 = YOLO(DEFAULT_DET_MODEL_ID_yolov8)
28
  current_model_id = model_id
29
 
 
23
  global current_model_id
24
  global model
25
  if model_id != current_model_id:
26
+ model = yolov5.load(model_id)
27
  # model_yolov8 = YOLO(DEFAULT_DET_MODEL_ID_yolov8)
28
  current_model_id = model_id
29