jedick commited on
Commit
592998f
·
1 Parent(s): e4c1af6

Use HF token for model download

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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(embedding_model_id, local_dir_use_symlinks=False)
 
 
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