Spaces:
Saad0KH
/
Running on Zero

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -122,7 +122,6 @@ def pil_to_binary_mask(pil_image, threshold=0):
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,6 +240,7 @@ def encode_image_to_base64(image):
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
 
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
  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