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

Fixed model download path

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