hysts HF staff yumingj commited on
Commit
c0e1f54
1 Parent(s): 0a27377

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>

Files changed (1) hide show
  1. model.py +2 -2
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('hysts/Text2Human',
69
- 'orig/pretrained_models.zip',
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: