Spaces:
Running
Running
Upload path to pretrained model on SHHQ (#7)
Browse files- Upload path to pretrained model on SHHQ (98acc39bafd230cd52cf9a326903701691e64549)
Co-authored-by: Yuming Jiang <yumingj@users.noreply.huggingface.co>
model.py
CHANGED
@@ -65,8 +65,8 @@ class Model:
|
|
65 |
if model_dir.exists():
|
66 |
return
|
67 |
token = os.getenv('HF_TOKEN')
|
68 |
-
path = huggingface_hub.hf_hub_download('
|
69 |
-
'
|
70 |
use_auth_token=token)
|
71 |
model_dir.mkdir()
|
72 |
with zipfile.ZipFile(path) as f:
|
|
|
65 |
if model_dir.exists():
|
66 |
return
|
67 |
token = os.getenv('HF_TOKEN')
|
68 |
+
path = huggingface_hub.hf_hub_download('yumingj/Text2Human_SSHQ',
|
69 |
+
'pretrained_models.zip',
|
70 |
use_auth_token=token)
|
71 |
model_dir.mkdir()
|
72 |
with zipfile.ZipFile(path) as f:
|