Spaces:
Saad0KH
/
Running on Zero

Saad0KH commited on
Commit
822aef6
ยท
verified ยท
1 Parent(s): 004975c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -119,6 +119,7 @@ def pil_to_binary_mask(pil_image, threshold=0):
119
  output_mask = Image.fromarray(mask)
120
  return output_mask
121
 
 
122
  def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed):
123
  device = "cuda"
124
  openpose_model.preprocessor.body_estimation.model.to(device)
 
119
  output_mask = Image.fromarray(mask)
120
  return output_mask
121
 
122
+ @spaces.GPU
123
  def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed):
124
  device = "cuda"
125
  openpose_model.preprocessor.body_estimation.model.to(device)