sneha
commited on
Commit
•
38d62ce
1
Parent(s):
8b10fec
add access token
Browse files
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)
|