rlawjdghek commited on
Commit
f9d01c1
β€’
1 Parent(s): 92a1987

remove space

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -1,4 +1,4 @@
1
- import spaces
2
  import os
3
  import sys
4
  import time
@@ -20,7 +20,7 @@ from utils_stableviton import get_batch, get_mask_location, tensor2img
20
  PROJECT_ROOT = Path(__file__).absolute().parents[1].absolute()
21
  sys.path.insert(0, str(PROJECT_ROOT))
22
 
23
- # from preprocess.detectron2.projects.DensePose.apply_net_gradio import DensePose4Gradio
24
  from preprocess.humanparsing.run_parsing import Parsing
25
  from preprocess.openpose.run_openpose import OpenPose
26
  import apply_net
@@ -34,10 +34,10 @@ IMG_W = 384
34
  openpose_model_hd = OpenPose(0)
35
  openpose_model_hd.preprocessor.body_estimation.model.to('cuda')
36
  parsing_model_hd = Parsing(0)
37
- # densepose_model_hd = DensePose4Gradio(
38
- # cfg='preprocess/detectron2/projects/DensePose/configs/densepose_rcnn_R_50_FPN_s1x.yaml',
39
- # model='https://dl.fbaipublicfiles.com/densepose/densepose_rcnn_R_50_FPN_s1x/165712039/model_final_162be9.pkl',
40
- # )
41
 
42
  category_dict = ['upperbody', 'lowerbody', 'dress']
43
  category_dict_utils = ['upper_body', 'lower_body', 'dresses']
 
1
+ # import spaces
2
  import os
3
  import sys
4
  import time
 
20
  PROJECT_ROOT = Path(__file__).absolute().parents[1].absolute()
21
  sys.path.insert(0, str(PROJECT_ROOT))
22
 
23
+ from preprocess.detectron2.projects.DensePose.apply_net_gradio import DensePose4Gradio
24
  from preprocess.humanparsing.run_parsing import Parsing
25
  from preprocess.openpose.run_openpose import OpenPose
26
  import apply_net
 
34
  openpose_model_hd = OpenPose(0)
35
  openpose_model_hd.preprocessor.body_estimation.model.to('cuda')
36
  parsing_model_hd = Parsing(0)
37
+ densepose_model_hd = DensePose4Gradio(
38
+ cfg='preprocess/detectron2/projects/DensePose/configs/densepose_rcnn_R_50_FPN_s1x.yaml',
39
+ model='https://dl.fbaipublicfiles.com/densepose/densepose_rcnn_R_50_FPN_s1x/165712039/model_final_162be9.pkl',
40
+ )
41
 
42
  category_dict = ['upperbody', 'lowerbody', 'dress']
43
  category_dict_utils = ['upper_body', 'lower_body', 'dresses']