fix: move gdrive to huggingface hub

app.py

@@ -5,6 +5,7 @@ import wget
 import gradio as gr
 import numpy as np
 import gdown
+from huggingface_hub import hf_hub_download
 from argparse import Namespace
 try:
     import detectron2
@@ -21,23 +22,50 @@ if torch.cuda.is_available():
 else:
     device = torch.device("cpu")
 
-# download metadata
-zs_weight_path = 'datasets/metadata/lvis_v1_clip_a+cname.npy'
-if not os.path.exists(zs_weight_path):
-    wget.download("https://github.com/facebookresearch/Detic/raw/main/datasets/metadata/lvis_v1_clip_a+cname.npy", out=zs_weight_path)
+# # download metadata
+# zs_weight_path = 'datasets/metadata/lvis_v1_clip_a+cname.npy'
+# if not os.path.exists(zs_weight_path):
+#     wget.download("https://github.com/facebookresearch/Detic/raw/main/datasets/metadata/lvis_v1_clip_a+cname.npy", out=zs_weight_path)
 
-base_cat_mask_path = "datasets/metadata/lvis_v1_base_cat_mask.npy"
-if not os.path.exists(base_cat_mask_path):
-    wget.download("https://docs.google.com/uc?export=download&id=1CbSs5yeqMsWDkRSsIlB-ln_bXDv686rH", out=base_cat_mask_path)
+# base_cat_mask_path = "datasets/metadata/lvis_v1_base_cat_mask.npy"
+# if not os.path.exists(base_cat_mask_path):
+#     wget.download("https://docs.google.com/uc?export=download&id=1CbSs5yeqMsWDkRSsIlB-ln_bXDv686rH", out=base_cat_mask_path)
 
-lvis_train_cat_info_path = "datasets/metadata/lvis_v1_train_cat_info.json"
-if not os.path.exists(lvis_train_cat_info_path):
-    wget.download("https://docs.google.com/uc?export=download&id=17WmkAJYBK4xT-YkiXLcwIWmtfulSUtmO", out=lvis_train_cat_info_path)
+# lvis_train_cat_info_path = "datasets/metadata/lvis_v1_train_cat_info.json"
+# if not os.path.exists(lvis_train_cat_info_path):
+#     wget.download("https://docs.google.com/uc?export=download&id=17WmkAJYBK4xT-YkiXLcwIWmtfulSUtmO", out=lvis_train_cat_info_path)
 
-# download model
-model_path = "models/proxydet_swinb_w_inl.pth"
-if not os.path.exists(model_path):
-    gdown.download("https://docs.google.com/uc?export=download&id=17kUPoi-pEK7BlTBheGzWxe_DXJlg28qF", model_path)
+# # download model
+# model_path = "models/proxydet_swinb_w_inl.pth"
+# if not os.path.exists(model_path):
+#     gdown.download("https://docs.google.com/uc?export=download&id=17kUPoi-pEK7BlTBheGzWxe_DXJlg28qF", model_path)
+hf_hub_download(
+    repo_id="doublejtoh/proxydet_data",
+    filename="models/proxydet_swinb_w_inl.pth",
+    repo_type="model",
+    local_dir="./"
+)
+
+hf_hub_download(
+    repo_id="doublejtoh/proxydet_data",
+    filename="datasets/metadata/lvis_v1_base_cat_mask.npy",
+    repo_type="model",
+    local_dir="./"
+)
+
+hf_hub_download(
+    repo_id="doublejtoh/proxydet_data",
+    filename="datasets/metadata/lvis_v1_clip_a+cname.npy",
+    repo_type="model",
+    local_dir="./"
+)
+
+hf_hub_download(
+    repo_id="doublejtoh/proxydet_data",
+    filename="datasets/metadata/lvis_v1_train_cat_info.json",
+    repo_type="model",
+    local_dir="./"
+)
 
 args = Namespace(
     base_cat_threshold=0.9,