Pie31415 commited on
Commit
9941f21
1 Parent(s): a4fd448

updated app

Browse files
Files changed (1) hide show
  1. app.py +5 -7
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,