Spaces:
Runtime error
Runtime error
onescotch
commited on
Commit
·
d200058
1
Parent(s):
66b16f2
load model from module repo
Browse files
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'
|