Spaces:
Sleeping
Sleeping
VasithaTilakumara commited on
Commit Β·
679fcb5
1
Parent(s): 1e5d740
debug load model function in models class
Browse files- utils/models.py +2 -1
utils/models.py
CHANGED
|
@@ -32,10 +32,11 @@ def load_model(model_path: str = "app_best.joblib"):
|
|
| 32 |
filename=model_path
|
| 33 |
)
|
| 34 |
model = joblib.load(model_path_hf)
|
| 35 |
-
|
| 36 |
raise FileNotFoundError(
|
| 37 |
f"β Model file not found. Tried: {model_path_hf} and {model_path_hf}"
|
| 38 |
)
|
|
|
|
| 39 |
|
| 40 |
# ------------------------------------------------------------
|
| 41 |
# π Example placeholder metrics and visuals
|
|
|
|
| 32 |
filename=model_path
|
| 33 |
)
|
| 34 |
model = joblib.load(model_path_hf)
|
| 35 |
+
|
| 36 |
raise FileNotFoundError(
|
| 37 |
f"β Model file not found. Tried: {model_path_hf} and {model_path_hf}"
|
| 38 |
)
|
| 39 |
+
return model
|
| 40 |
|
| 41 |
# ------------------------------------------------------------
|
| 42 |
# π Example placeholder metrics and visuals
|