levihsu commited on
Commit
c1569cf
1 Parent(s): 5533ab7

Update run/gradio_ootd.py

Browse files
Files changed (1) hide show
  1. run/gradio_ootd.py +4 -3
run/gradio_ootd.py CHANGED
@@ -20,9 +20,9 @@ openpose_model_hd = OpenPose(0)
20
  parsing_model_hd = Parsing(0)
21
  ootd_model_hd = OOTDiffusionHD(0)
22
 
23
- openpose_model_dc = OpenPose(1)
24
- parsing_model_dc = Parsing(1)
25
- ootd_model_dc = OOTDiffusionDC(1)
26
 
27
 
28
  category_dict = ['upperbody', 'lowerbody', 'dress']
@@ -70,6 +70,7 @@ def process_hd(vton_img, garm_img, n_samples, n_steps, image_scale, seed):
70
 
71
  return images
72
 
 
73
  def process_dc(vton_img, garm_img, category, n_samples, n_steps, image_scale, seed):
74
  model_type = 'dc'
75
  if category == 'Upper-body':
 
20
  parsing_model_hd = Parsing(0)
21
  ootd_model_hd = OOTDiffusionHD(0)
22
 
23
+ openpose_model_dc = OpenPose(0)
24
+ parsing_model_dc = Parsing(0)
25
+ ootd_model_dc = OOTDiffusionDC(0)
26
 
27
 
28
  category_dict = ['upperbody', 'lowerbody', 'dress']
 
70
 
71
  return images
72
 
73
+ @spaces.GPU
74
  def process_dc(vton_img, garm_img, category, n_samples, n_steps, image_scale, seed):
75
  model_type = 'dc'
76
  if category == 'Upper-body':