updated app

app.py

@@ -1,18 +1,17 @@
+import sys
+import torch
 import gradio as gr
-from rome.infer import Infer
 import pickle
+
 from easydict import EasyDict as edict
 from huggingface_hub import hf_hub_download
+
+from rome.infer import Infer
 from rome.src.utils.processing import process_black_shape, tensor2image
-from rome.src.utils import args as args_utils
-import sys
-import torch
-import torch
 
 sys.path.append("./rome/")
 sys.path.append('./DECA')
 
-
 # loading models ---- create model repo
 
 default_modnet_path = hf_hub_download(
@@ -20,7 +19,6 @@ default_modnet_path = hf_hub_download(
 default_model_path = hf_hub_download('Pie31415/rome', 'rome.pth')
 
 # parser configurations
-
 args = edict({
     "save_dir": ".",
     "save_render": True,