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