haseebnawazz commited on
Commit
233d82c
·
verified ·
1 Parent(s): 10f87bb

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +1 -1
api.py CHANGED
@@ -14,7 +14,7 @@ app = FastAPI()
14
  model_path = hf_hub_download(
15
  repo_id="haseebnawazz/sleep-stage-classifier-model", # Update with your HF model repo
16
  filename="model.joblib",
17
- cache_dir="/app/hf_cache" # Ensure the cache is stored in a writable directory
18
  )
19
  model = joblib.load(model_path)
20
 
 
14
  model_path = hf_hub_download(
15
  repo_id="haseebnawazz/sleep-stage-classifier-model", # Update with your HF model repo
16
  filename="model.joblib",
17
+ cache_dir="/tmp/hf_cache" # Ensure the cache is stored in a writable directory
18
  )
19
  model = joblib.load(model_path)
20