atatakun commited on
Commit
bbce04f
·
1 Parent(s): b58b18c

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -217,7 +217,7 @@ model_inpaint = None
217
  def inpaint(image):
218
  # image = resize_image(img, res)
219
  color = HWC3(image["image"])
220
- alpha = HWC3(image["mask"][:, :, 0:1])
221
  result = np.concatenate([color, alpha], axis=2)
222
  return [result]
223
 
 
217
  def inpaint(image):
218
  # image = resize_image(img, res)
219
  color = HWC3(image["image"])
220
+ alpha = image["mask"][:, :, 1]
221
  result = np.concatenate([color, alpha], axis=2)
222
  return [result]
223