doublejtoh commited on
Commit
3ad7c50
·
1 Parent(s): d9cdf23

fix: move gdrive to huggingface hub

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -39,6 +39,7 @@ else:
39
  # model_path = "models/proxydet_swinb_w_inl.pth"
40
  # if not os.path.exists(model_path):
41
  # gdown.download("https://docs.google.com/uc?export=download&id=17kUPoi-pEK7BlTBheGzWxe_DXJlg28qF", model_path)
 
42
  hf_hub_download(
43
  repo_id="doublejtoh/proxydet_data",
44
  filename="models/proxydet_swinb_w_inl.pth",
@@ -66,6 +67,7 @@ hf_hub_download(
66
  repo_type="model",
67
  local_dir="./"
68
  )
 
69
 
70
  args = Namespace(
71
  base_cat_threshold=0.9,
 
39
  # model_path = "models/proxydet_swinb_w_inl.pth"
40
  # if not os.path.exists(model_path):
41
  # gdown.download("https://docs.google.com/uc?export=download&id=17kUPoi-pEK7BlTBheGzWxe_DXJlg28qF", model_path)
42
+
43
  hf_hub_download(
44
  repo_id="doublejtoh/proxydet_data",
45
  filename="models/proxydet_swinb_w_inl.pth",
 
67
  repo_type="model",
68
  local_dir="./"
69
  )
70
+ model_path = "models/proxydet_swinb_w_inl.pth"
71
 
72
  args = Namespace(
73
  base_cat_threshold=0.9,