Spaces:
Running
on
Zero
Running
on
Zero
jedick
commited on
Commit
·
592998f
1
Parent(s):
e4c1af6
Use HF token for model download
Browse files
app.py
CHANGED
|
@@ -22,10 +22,12 @@ load_dotenv(dotenv_path=".env", override=True)
|
|
| 22 |
|
| 23 |
# Download model snapshots from Hugging Face Hub
|
| 24 |
print(f"Downloading checkpoints for {model_id}...")
|
| 25 |
-
ckpt_dir = snapshot_download(model_id, local_dir_use_symlinks=False)
|
| 26 |
print(f"Using checkpoints from {ckpt_dir}")
|
| 27 |
print(f"Downloading checkpoints for {embedding_model_id}...")
|
| 28 |
-
embedding_ckpt_dir = snapshot_download(
|
|
|
|
|
|
|
| 29 |
print(f"Using embedding checkpoints from {embedding_ckpt_dir}")
|
| 30 |
|
| 31 |
# Download and extract data if data directory is not present
|
|
|
|
| 22 |
|
| 23 |
# Download model snapshots from Hugging Face Hub
|
| 24 |
print(f"Downloading checkpoints for {model_id}...")
|
| 25 |
+
ckpt_dir = snapshot_download(model_id, local_dir_use_symlinks=False, token=True)
|
| 26 |
print(f"Using checkpoints from {ckpt_dir}")
|
| 27 |
print(f"Downloading checkpoints for {embedding_model_id}...")
|
| 28 |
+
embedding_ckpt_dir = snapshot_download(
|
| 29 |
+
embedding_model_id, local_dir_use_symlinks=False, token=True
|
| 30 |
+
)
|
| 31 |
print(f"Using embedding checkpoints from {embedding_ckpt_dir}")
|
| 32 |
|
| 33 |
# Download and extract data if data directory is not present
|