Spaces:
Sleeping
Sleeping
Stanislav
commited on
Commit
·
46cdc8b
1
Parent(s):
1d98d28
feat: cashdir
Browse files- run_fastapi.py +1 -0
run_fastapi.py
CHANGED
|
@@ -36,6 +36,7 @@ if not os.path.exists(MODEL_PATH):
|
|
| 36 |
cached_path = hf_hub_download(
|
| 37 |
repo_id=REPO_ID,
|
| 38 |
filename=FILENAME,
|
|
|
|
| 39 |
local_dir_use_symlinks=False
|
| 40 |
)
|
| 41 |
shutil.copy(cached_path, MODEL_PATH)
|
|
|
|
| 36 |
cached_path = hf_hub_download(
|
| 37 |
repo_id=REPO_ID,
|
| 38 |
filename=FILENAME,
|
| 39 |
+
cache_dir="./weights/.hf_cache",
|
| 40 |
local_dir_use_symlinks=False
|
| 41 |
)
|
| 42 |
shutil.copy(cached_path, MODEL_PATH)
|