leo-bourrel commited on
Commit
d304ca0
·
1 Parent(s): d4e4cf4

feat: update folder path

Browse files
Files changed (1) hide show
  1. src/yolo_predictions.py +1 -1
src/yolo_predictions.py CHANGED
@@ -55,7 +55,7 @@ def load_model(model_file):
55
 
56
  try:
57
  print(f"Loading YOLO model from file: {model_file}...")
58
- model = YOLO(model_file)
59
  print(f"Model loaded successfully!")
60
  return model
61
  except Exception as e:
 
55
 
56
  try:
57
  print(f"Loading YOLO model from file: {model_file}...")
58
+ model = YOLO(f"models/{model_file}")
59
  print(f"Model loaded successfully!")
60
  return model
61
  except Exception as e: