Spaces:
Saad0KH
/
Running on Zero

Saad0KH commited on
Commit
6153abe
ยท
verified ยท
1 Parent(s): 2b45835

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -122,6 +122,7 @@ def pil_to_binary_mask(pil_image, threshold=0):
122
  output_mask = Image.fromarray(mask)
123
  return output_mask
124
 
 
125
  def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed):
126
  device = "cuda"
127
  openpose_model.preprocessor.body_estimation.model.to(device)
@@ -240,7 +241,6 @@ def encode_image_to_base64(image):
240
  image.save(buffered, format="PNG")
241
  return base64.b64encode(buffered.getvalue()).decode('utf-8')
242
 
243
- @spaces.GPU
244
  @app.route('/tryon', methods=['POST'])
245
  def tryon():
246
  data = request.json
 
122
  output_mask = Image.fromarray(mask)
123
  return output_mask
124
 
125
+ @spaces.GPU
126
  def start_tryon(dict, garm_img, garment_des, is_checked, is_checked_crop, denoise_steps, seed):
127
  device = "cuda"
128
  openpose_model.preprocessor.body_estimation.model.to(device)
 
241
  image.save(buffered, format="PNG")
242
  return base64.b64encode(buffered.getvalue()).decode('utf-8')
243
 
 
244
  @app.route('/tryon', methods=['POST'])
245
  def tryon():
246
  data = request.json