Spaces:
Running
Running
Update
Browse files
model.py
CHANGED
@@ -64,10 +64,8 @@ class Model:
|
|
64 |
model_dir = pathlib.Path('pretrained_models')
|
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:
|
73 |
f.extractall(model_dir)
|
|
|
64 |
model_dir = pathlib.Path('pretrained_models')
|
65 |
if model_dir.exists():
|
66 |
return
|
|
|
67 |
path = huggingface_hub.hf_hub_download('yumingj/Text2Human_SSHQ',
|
68 |
+
'pretrained_models.zip')
|
|
|
69 |
model_dir.mkdir()
|
70 |
with zipfile.ZipFile(path) as f:
|
71 |
f.extractall(model_dir)
|