minchul
/

cvlface_arcface_ir101_webface4m

Feature Extraction

Model card Files Files and versions Community

minchul commited on Jun 10

Commit

95922ef

•

1 Parent(s): 87b3115

Upload directory

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -39,6 +39,7 @@ import torch
 # helpfer function to download huggingface repo and use model
 def download(repo_id, path, HF_TOKEN=None):
     files_path = os.path.join(path, 'files.txt')
     if not os.path.exists(files_path):
         hf_hub_download(repo_id, 'files.txt', token=HF_TOKEN, local_dir=path, local_dir_use_symlinks=False)

 # helpfer function to download huggingface repo and use model
 def download(repo_id, path, HF_TOKEN=None):
+    os.makedirs(path, exist_ok=True)
     files_path = os.path.join(path, 'files.txt')
     if not os.path.exists(files_path):
         hf_hub_download(repo_id, 'files.txt', token=HF_TOKEN, local_dir=path, local_dir_use_symlinks=False)