Pie31415 commited on
Commit
75bf09d
1 Parent(s): c1eb1ed

updated app

Browse files
Files changed (2) hide show
  1. .gitmodules +2 -2
  2. app.py +3 -3
.gitmodules CHANGED
@@ -1,5 +1,5 @@
1
- [submodule "ROME"]
2
- path = ROME
3
  url = https://github.com/SamsungLabs/rome
4
  [submodule "DECA"]
5
  path = DECA
 
1
+ [submodule "rome"]
2
+ path = rome
3
  url = https://github.com/SamsungLabs/rome
4
  [submodule "DECA"]
5
  path = DECA
app.py CHANGED
@@ -5,9 +5,9 @@ import torch
5
  import matplotlib.pyplot as plt
6
  from PIL import Image
7
 
8
- sys.path.append("./ROME")
9
- from ROME.src.utils import args as args_utils
10
- from ROME.src.utils.processing import process_black_shape, tensor2image
11
 
12
  # loading models ---- create model repo
13
  from huggingface_hub import hf_hub_url
 
5
  import matplotlib.pyplot as plt
6
  from PIL import Image
7
 
8
+ sys.path.append("./rome")
9
+ from rome.src.utils import args as args_utils
10
+ from rome.src.utils.processing import process_black_shape, tensor2image
11
 
12
  # loading models ---- create model repo
13
  from huggingface_hub import hf_hub_url