Upload folder using huggingface_hub

demo.py

@@ -12,7 +12,7 @@ import os, cv2, time, math
 print("=> Loading libraries...")
 start = time.time()
 
-import requests, torch
+import requests, torch, argparse
 import gradio as gr
 from torchvision import transforms
 from datasets import load_dataset
@@ -21,6 +21,13 @@ from timm.models import create_model, load_checkpoint
 from pytorch_grad_cam import GradCAM
 from pytorch_grad_cam.utils.image import show_cam_on_image
 
+parser = argparse.ArgumentParser()
+parser.add_argument("--local", action='store_true')
+args = parser.parse_args()
+
+if not args.local:
+    from huggingface_hub import login
+    login(token=os.environ["HF_TOKEN"])
 
 device = "cuda" if torch.cuda.is_available() else "cpu"