Upload directory

README.md

@@ -69,11 +69,18 @@ def load_model_by_repo_id(repo_id, save_path, HF_TOKEN=None, force_download=Fals
 
 
 if __name__ == '__main__':
+    
     HF_TOKEN = 'YOUR_HUGGINGFACE_TOKEN'
     path = 'path/to/store/model/locally'
     repo_id = 'minchul/cvlface_arcface_ir101_webface4m'
     model = load_model_by_repo_id(repo_id, path, HF_TOKEN)
-    input = torch.randn(1, 3, 112, 112)
+
+    # input is a rgb image normalized.
+    from torchvision.transforms import Compose, ToTensor, Normalize
+    from PIL import Image
+    img = Image.open('path/to/image.jpg')
+    trans = Compose([ToTensor(), Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])])
+    input = trans(img).unsqueeze(0)  # torch.randn(1, 3, 112, 112)
     out = model(input)
 ```