Spaces:
Running
Running
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)
|