levihsu commited on
Commit
57ef03e
β€’
1 Parent(s): a657367

Update run/gradio_ootd.py

Browse files
Files changed (1) hide show
  1. run/gradio_ootd.py +1 -1
run/gradio_ootd.py CHANGED
@@ -36,7 +36,7 @@ garment_hd = os.path.join(example_path, 'garment/03244_00.jpg')
36
  model_dc = os.path.join(example_path, 'model/model_8.png')
37
  garment_dc = os.path.join(example_path, 'garment/048554_1.jpg')
38
 
39
- +@spaces.GPU
40
  def process_hd(vton_img, garm_img, n_samples, n_steps, image_scale, seed):
41
  model_type = 'hd'
42
  category = 0 # 0:upperbody; 1:lowerbody; 2:dress
 
36
  model_dc = os.path.join(example_path, 'model/model_8.png')
37
  garment_dc = os.path.join(example_path, 'garment/048554_1.jpg')
38
 
39
+ @spaces.GPU
40
  def process_hd(vton_img, garm_img, n_samples, n_steps, image_scale, seed):
41
  model_type = 'hd'
42
  category = 0 # 0:upperbody; 1:lowerbody; 2:dress