atatakun commited on
Commit
a8a29d0
1 Parent(s): c6aaa84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -228,16 +228,17 @@ model_inpaint = None
228
  def inpaint(image, invert):
229
  # image = resize_image(img, res)
230
  # color = HWC3(image["image"])
231
- color = HWC3(image["background"])
232
  # if(invert):
233
  # alpha = image["mask"][:, :, 0:1]
234
- alpha = image["layers"][0]
235
  # else:
236
  # alpha = 255 - image["mask"][:, :, 0:1]
237
  # alpha = 255 - image["layers"][0]
238
  # result = np.concatenate([color, alpha], axis=2)
239
- result = alpha
240
- return [result]
 
241
 
242
 
243
  def predict(im):
 
228
  def inpaint(image, invert):
229
  # image = resize_image(img, res)
230
  # color = HWC3(image["image"])
231
+ # color = HWC3(image["background"])
232
  # if(invert):
233
  # alpha = image["mask"][:, :, 0:1]
234
+ # alpha = image["layers"][0]
235
  # else:
236
  # alpha = 255 - image["mask"][:, :, 0:1]
237
  # alpha = 255 - image["layers"][0]
238
  # result = np.concatenate([color, alpha], axis=2)
239
+ # result = alpha
240
+ # return [result]
241
+ return [image]
242
 
243
 
244
  def predict(im):