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