multimodalart HF staff commited on
Commit
82145d6
1 Parent(s): c457a0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -105,9 +105,11 @@ hf_hub_download(
105
  local_dir="/data/checkpoints",
106
  )
107
  # download antelopev2
108
- if not os.path.exists("/data/antelopev2.zip"):
109
- gdown.download(url="https://drive.google.com/file/d/18wEUfMNohBJ4K3Ly5wpTejPfDzp-8fI8/view?usp=sharing", output="/data/", quiet=False, fuzzy=True)
110
- os.system("unzip /data/antelopev2.zip -d /data/models/")
 
 
111
 
112
  app = FaceAnalysis(name='antelopev2', root='/data', providers=['CPUExecutionProvider'])
113
  app.prepare(ctx_id=0, det_size=(640, 640))
 
105
  local_dir="/data/checkpoints",
106
  )
107
  # download antelopev2
108
+ #if not os.path.exists("/data/antelopev2.zip"):
109
+ # gdown.download(url="https://drive.google.com/file/d/18wEUfMNohBJ4K3Ly5wpTejPfDzp-8fI8/view?usp=sharing", output="/data/", quiet=False, fuzzy=True)
110
+ # os.system("unzip /data/antelopev2.zip -d /data/models/")
111
+
112
+ snapshot_download(repo_id="https://huggingface.co/DIAMONIK7777/antelopev2", local_dir="/data/models")
113
 
114
  app = FaceAnalysis(name='antelopev2', root='/data', providers=['CPUExecutionProvider'])
115
  app.prepare(ctx_id=0, det_size=(640, 640))