Hugging Face's logo

Spaces:
1vash
/
demo-flask-docker-template
Runtime error

App Files Community
Ivan Shelonik commited on
Commit
75241f1
·
1 Parent(s): 07f38a9

upd: cache dir

Browse files
Files changed (1) hide show
  1. api_server.py +1 -1
api_server.py CHANGED
@@ -30,7 +30,7 @@ elif load_type == 'remote_hub_download':
30
  model = keras.models.load_model(hf_hub_download(repo_id=REPO_ID, filename="saved_model.pb"))
31
  elif load_type == 'remote_hub_from_pretrained':
32
  from huggingface_hub import from_pretrained_keras
33
- model = from_pretrained_keras(REPO_ID, cache_dir='./artifacts')
34
 
35
 
36
  # Initialize the Flask application
 
30
  model = keras.models.load_model(hf_hub_download(repo_id=REPO_ID, filename="saved_model.pb"))
31
  elif load_type == 'remote_hub_from_pretrained':
32
  from huggingface_hub import from_pretrained_keras
33
+ model = from_pretrained_keras(REPO_ID, cache_dir='./artifacts/')
34
 
35
 
36
  # Initialize the Flask application