sneha commited on
Commit
38d62ce
1 Parent(s): 8b10fec

add access token

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -11,6 +11,8 @@ import vc_models
11
  #import eaif_models
12
  import torchvision
13
 
 
 
14
  eai_filepath = vc_models.__file__.split('src')[0]
15
  MODEL_DIR=os.path.join(os.path.dirname(eai_filepath),'model_ckpts')
16
  if not os.path.isdir(MODEL_DIR):
@@ -25,7 +27,7 @@ def get_model():
25
  download_bin()
26
  if MODEL_TUPLE is None:
27
  model_cfg = omegaconf.OmegaConf.load(
28
- hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
29
  )
30
  model_cfg['model']['checkpoint_path'] = None
31
  model_cfg['model']['checkpoint_path'] = 'model_ckpts/vc1_vitb.pth'#os.path.join(os.getcwd(),'pytorch_model.bin')
@@ -41,7 +43,7 @@ def download_bin():
41
  print(bin_path)
42
  if not os.path.isfile(bin_path):
43
  #with open(bin_file,'w') as f:
44
- model_bin = hf_hub_download(repo_id=REPO_ID, filename='pytorch_model.bin',local_dir=MODEL_DIR,local_dir_use_symlinks=True)
45
  os.rename(model_bin, bin_path)
46
  print(type(model_bin))
47
  # os.rename(model_bin, bin_file)
 
11
  #import eaif_models
12
  import torchvision
13
 
14
+
15
+ HF_TOKEN = os.environ['HF_ACC_TOKEN']
16
  eai_filepath = vc_models.__file__.split('src')[0]
17
  MODEL_DIR=os.path.join(os.path.dirname(eai_filepath),'model_ckpts')
18
  if not os.path.isdir(MODEL_DIR):
 
27
  download_bin()
28
  if MODEL_TUPLE is None:
29
  model_cfg = omegaconf.OmegaConf.load(
30
+ hf_hub_download(repo_id=REPO_ID, filename=FILENAME,use_access_token=HF_TOKEN)
31
  )
32
  model_cfg['model']['checkpoint_path'] = None
33
  model_cfg['model']['checkpoint_path'] = 'model_ckpts/vc1_vitb.pth'#os.path.join(os.getcwd(),'pytorch_model.bin')
 
43
  print(bin_path)
44
  if not os.path.isfile(bin_path):
45
  #with open(bin_file,'w') as f:
46
+ model_bin = hf_hub_download(repo_id=REPO_ID, filename='pytorch_model.bin',local_dir=MODEL_DIR,local_dir_use_symlinks=True,use_access_token=HF_TOKEN)
47
  os.rename(model_bin, bin_path)
48
  print(type(model_bin))
49
  # os.rename(model_bin, bin_file)