upd: cache_dirs

api_server.py

@@ -30,7 +30,7 @@ elif load_type == 'remote_hub_download':
     model = keras.models.load_model(hf_hub_download(repo_id=REPO_ID, filename="saved_model.pb"))
 elif load_type == 'remote_hub_from_pretrained':
     from huggingface_hub import from_pretrained_keras
-    model = from_pretrained_keras(REPO_ID)
+    model = from_pretrained_keras(REPO_ID, cache_dir='./artifacts')
 
 
 # Initialize the Flask application