hongfz16 commited on
Commit
374ae12
1 Parent(s): add49e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -46,7 +46,7 @@ def download_pretrained_models():
46
  # print('Google Drive download failed.\n' \
47
  # 'Trying do download from alternate server')
48
  # download_file(session, eva3d_deepfashion_model, use_alt_url=True)
49
- eva3d_ckpt = hf_hub_download(repo_id="hongfz16/EVA3D", filename="models_0420000.pt")
50
  os.system("mkdir -p checkpoint/512x256_deepfashion/volume_renderer")
51
  os.system("mkdir -p smpl_models/smpl")
52
  os.system(f"cp {eva3d_ckpt} checkpoint/512x256_deepfashion/volume_renderer/models_0420000.pt")
@@ -58,7 +58,7 @@ def download_pretrained_models():
58
  # print('Google Drive download failed.\n' \
59
  # 'Trying do download from alternate server')
60
  # download_file(session, smpl_model, use_alt_url=True)
61
- smpl_pkl = hf_hub_download(repo_id="hongfz16/EVA3D", filename="SMPL_NEUTRAL.pkl")
62
  os.system(f"cp {smpl_pkl} smpl_models/smpl/SMPL_NEUTRAL.pkl")
63
 
64
  download_pretrained_models()
 
46
  # print('Google Drive download failed.\n' \
47
  # 'Trying do download from alternate server')
48
  # download_file(session, eva3d_deepfashion_model, use_alt_url=True)
49
+ eva3d_ckpt = hf_hub_download(repo_id="hongfz16/EVA3D", filename="models_0420000.pt", token=os.environ['hf_token'])
50
  os.system("mkdir -p checkpoint/512x256_deepfashion/volume_renderer")
51
  os.system("mkdir -p smpl_models/smpl")
52
  os.system(f"cp {eva3d_ckpt} checkpoint/512x256_deepfashion/volume_renderer/models_0420000.pt")
 
58
  # print('Google Drive download failed.\n' \
59
  # 'Trying do download from alternate server')
60
  # download_file(session, smpl_model, use_alt_url=True)
61
+ smpl_pkl = hf_hub_download(repo_id="hongfz16/EVA3D", filename="SMPL_NEUTRAL.pkl", token=os.environ['hf_token'])
62
  os.system(f"cp {smpl_pkl} smpl_models/smpl/SMPL_NEUTRAL.pkl")
63
 
64
  download_pretrained_models()