SaiStack commited on
Commit
d4536af
·
1 Parent(s): a858de9

Fixed model download path

Browse files
Files changed (1) hide show
  1. app/pipelines/main_pipeline.py +1 -1
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="model"
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)