Upload directory

README.md

@@ -68,9 +68,14 @@ def load_model_by_repo_id(repo_id, save_path, HF_TOKEN=None, force_download=Fals
     download(repo_id, save_path, HF_TOKEN)
     return load_model_from_local_path(save_path, HF_TOKEN)
 
+
+if __name__ == '__main__':
     
     # load model
-    aligner = load_model_by_repo_id(repo_id, path, HF_TOKEN, force_download=False)
+    HF_TOKEN = 'YOUR_HUGGINGFACE_TOKEN'
+    path = 'path/to/store/model/locally'
+    repo_id = 'minchul/cvlface_DFA_resnet50'
+    aligner = load_model_by_repo_id(repo_id, path, HF_TOKEN)
 
     # input is a rgb image normalized.
     from torchvision.transforms import Compose, ToTensor, Normalize