Spaces:
Sleeping
Sleeping
Create download_mode.py
Browse files- download_mode.py +8 -0
download_mode.py
ADDED
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from huggingface_hub import hf_hub_download
|
2 |
+
import os
|
3 |
+
|
4 |
+
repo_id = os.environ.get("MODEL_REPO")
|
5 |
+
filename = os.environ.get("MODEL_FILE")
|
6 |
+
|
7 |
+
print("Downloading model:", repo_id, filename)
|
8 |
+
hf_hub_download(repo_id=repo_id, filename=filename, local_dir=".", local_dir_use_symlinks=False)
|