Spaces:
Build error
Build error
updated app
Browse files
app.py
CHANGED
@@ -1,18 +1,17 @@
|
|
|
|
|
|
1 |
import gradio as gr
|
2 |
-
from rome.infer import Infer
|
3 |
import pickle
|
|
|
4 |
from easydict import EasyDict as edict
|
5 |
from huggingface_hub import hf_hub_download
|
|
|
|
|
6 |
from rome.src.utils.processing import process_black_shape, tensor2image
|
7 |
-
from rome.src.utils import args as args_utils
|
8 |
-
import sys
|
9 |
-
import torch
|
10 |
-
import torch
|
11 |
|
12 |
sys.path.append("./rome/")
|
13 |
sys.path.append('./DECA')
|
14 |
|
15 |
-
|
16 |
# loading models ---- create model repo
|
17 |
|
18 |
default_modnet_path = hf_hub_download(
|
@@ -20,7 +19,6 @@ default_modnet_path = hf_hub_download(
|
|
20 |
default_model_path = hf_hub_download('Pie31415/rome', 'rome.pth')
|
21 |
|
22 |
# parser configurations
|
23 |
-
|
24 |
args = edict({
|
25 |
"save_dir": ".",
|
26 |
"save_render": True,
|
|
|
1 |
+
import sys
|
2 |
+
import torch
|
3 |
import gradio as gr
|
|
|
4 |
import pickle
|
5 |
+
|
6 |
from easydict import EasyDict as edict
|
7 |
from huggingface_hub import hf_hub_download
|
8 |
+
|
9 |
+
from rome.infer import Infer
|
10 |
from rome.src.utils.processing import process_black_shape, tensor2image
|
|
|
|
|
|
|
|
|
11 |
|
12 |
sys.path.append("./rome/")
|
13 |
sys.path.append('./DECA')
|
14 |
|
|
|
15 |
# loading models ---- create model repo
|
16 |
|
17 |
default_modnet_path = hf_hub_download(
|
|
|
19 |
default_model_path = hf_hub_download('Pie31415/rome', 'rome.pth')
|
20 |
|
21 |
# parser configurations
|
|
|
22 |
args = edict({
|
23 |
"save_dir": ".",
|
24 |
"save_render": True,
|