onescotch commited on
Commit
d200058
1 Parent(s): 66b16f2

load model from module repo

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,12 +7,12 @@ import gradio as gr
7
  import torch
8
  import math
9
  import spaces
10
-
11
  try:
12
  import mmpose
13
  except:
14
  os.system('pip install /home/user/app/main/transformer_utils')
15
-
16
  os.system('cp -rf /home/user/app/assets/conversions.py /home/user/.pyenv/versions/3.9.18/lib/python3.9/site-packages/torchgeometry/core/conversions.py')
17
  DEFAULT_MODEL='smpler_x_h32'
18
  OUT_FOLDER = '/home/user/app/demo_out'
 
7
  import torch
8
  import math
9
  import spaces
10
+ from huggingface_hub import hf_hub_download
11
  try:
12
  import mmpose
13
  except:
14
  os.system('pip install /home/user/app/main/transformer_utils')
15
+ hf_hub_download(repo_id="caizhongang/SMPLer-X", filename="smpler_x_h32.pth.tar", local_dir="/home/user/app/pretrained_models")
16
  os.system('cp -rf /home/user/app/assets/conversions.py /home/user/.pyenv/versions/3.9.18/lib/python3.9/site-packages/torchgeometry/core/conversions.py')
17
  DEFAULT_MODEL='smpler_x_h32'
18
  OUT_FOLDER = '/home/user/app/demo_out'