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
|