Spaces:
Running
Running
Update utils/model_loader.py
Browse files- utils/model_loader.py +1 -1
utils/model_loader.py
CHANGED
|
@@ -37,7 +37,7 @@ def load_model_and_config():
|
|
| 37 |
model.load_state_dict(state_dict)
|
| 38 |
model.eval()
|
| 39 |
else:
|
| 40 |
-
print(f"
|
| 41 |
exit()
|
| 42 |
|
| 43 |
return {
|
|
|
|
| 37 |
model.load_state_dict(state_dict)
|
| 38 |
model.eval()
|
| 39 |
else:
|
| 40 |
+
print(f"Model file not found at {MODEL_PATH}")
|
| 41 |
exit()
|
| 42 |
|
| 43 |
return {
|