Spaces:
Saad0KH
/
Running on Zero

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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -246,6 +246,7 @@ def tryon():
246
  data = request.json
247
 
248
  human_image = decode_image_from_base64(data['human_image'])
 
249
  garment_image = decode_image_from_base64(data['garment_image'])
250
  description = data.get('description')
251
  use_auto_mask = data.get('use_auto_mask', True)
@@ -255,7 +256,7 @@ def tryon():
255
 
256
  human_dict = {
257
  'background': human_image,
258
- 'layers': [human_image] if not use_auto_mask else None,
259
  'composite': None
260
  }
261
 
 
246
  data = request.json
247
 
248
  human_image = decode_image_from_base64(data['human_image'])
249
+ mask_image = decode_image_from_base64(data['mask_image'])
250
  garment_image = decode_image_from_base64(data['garment_image'])
251
  description = data.get('description')
252
  use_auto_mask = data.get('use_auto_mask', True)
 
256
 
257
  human_dict = {
258
  'background': human_image,
259
+ 'layers': [mask_image] if not use_auto_mask else None,
260
  'composite': None
261
  }
262