Pie31415 commited on
Commit
c11ea8c
1 Parent(s): ab66a38

resolved import issues

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,12 +6,12 @@ import pickle
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
  default_modnet_path = hf_hub_download('Pie31415/rome', 'modnet_photographic_portrait_matting.ckpt')
17
  default_model_path = hf_hub_download('Pie31415/rome', 'rome.pth')
 
6
  from easydict import EasyDict as edict
7
  from huggingface_hub import hf_hub_download
8
 
 
 
 
9
  sys.path.append("./rome/")
10
  sys.path.append('./DECA')
11
 
12
+ from rome.infer import Infer
13
+ from rome.src.utils.processing import process_black_shape, tensor2image
14
+
15
  # loading models ---- create model repo
16
  default_modnet_path = hf_hub_download('Pie31415/rome', 'modnet_photographic_portrait_matting.ckpt')
17
  default_model_path = hf_hub_download('Pie31415/rome', 'rome.pth')