Update app.py

app.py

@@ -228,15 +228,14 @@ model_inpaint = None
 def inpaint(image, invert):
 #    image = resize_image(img, res)
 #    color = HWC3(image["image"])
-    color = HWC3(image["composite"])
+    color = HWC3(image["backbround"])
     if(invert):
 #        alpha = image["mask"][:, :, 0:1]
-        alpha = image["background"][:, :, 0:1]
+        alpha = image["layers"][0][:, :, 0:1]
     else:
 #        alpha = 255 - image["mask"][:, :, 0:1]
-        alpha = 255 - image["background"][:, :, 0:1]
-#    result = np.concatenate([color, alpha], axis=2)
-    result = np.concatenate(color)
+        alpha = 255 - image["layers"][0][:, :, 0:1]
+    result = np.concatenate([color, alpha], axis=2)
     return [result]