Spaces:
Paused
Paused
Fixed model download path
Browse files
app/pipelines/main_pipeline.py
CHANGED
|
@@ -207,7 +207,8 @@ def load_model(model_key):
|
|
| 207 |
model_path = hf_hub_download(
|
| 208 |
repo_id=hf_repo_id,
|
| 209 |
filename=f"{model_key}.keras",
|
| 210 |
-
repo_type="model"
|
|
|
|
| 211 |
)
|
| 212 |
|
| 213 |
model = keras_load_model(model_path)
|
|
|
|
| 207 |
model_path = hf_hub_download(
|
| 208 |
repo_id=hf_repo_id,
|
| 209 |
filename=f"{model_key}.keras",
|
| 210 |
+
repo_type="model",
|
| 211 |
+
cache_dir="model"
|
| 212 |
)
|
| 213 |
|
| 214 |
model = keras_load_model(model_path)
|