chenlei commited on
Commit
bfa0351
·
1 Parent(s): 74d6568
Files changed (1) hide show
  1. run/gradio_ootd.py +3 -3
run/gradio_ootd.py CHANGED
@@ -21,9 +21,9 @@ openpose_model_hd = OpenPose(0)
21
  parsing_model_hd = Parsing(0)
22
  ootd_model_hd = OOTDiffusionHD(0)
23
 
24
- openpose_model_dc = OpenPose(1)
25
- parsing_model_dc = Parsing(1)
26
- ootd_model_dc = OOTDiffusionDC(1)
27
 
28
 
29
  category_dict = ['upperbody', 'lowerbody', 'dress']
 
21
  parsing_model_hd = Parsing(0)
22
  ootd_model_hd = OOTDiffusionHD(0)
23
 
24
+ openpose_model_dc = OpenPose(0)
25
+ parsing_model_dc = Parsing(0)
26
+ ootd_model_dc = OOTDiffusionDC(0)
27
 
28
 
29
  category_dict = ['upperbody', 'lowerbody', 'dress']