Spaces:
Sleeping
Sleeping
Commit
·
d304ca0
1
Parent(s):
d4e4cf4
feat: update folder path
Browse files- 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:
|