Update app.py

app.py

@@ -231,10 +231,10 @@ def inpaint(image, invert):
     color = HWC3(image["background"])
     if(invert):
 #        alpha = image["mask"][:, :, 0:1]
-        alpha = image["layers"][0][:, :, 0:1]
+        alpha = image["layers"][0]
     else:
 #        alpha = 255 - image["mask"][:, :, 0:1]
-        alpha = 255 - image["layers"][0][:, :, 0:1]
+        alpha = 255 - image["layers"][0]
 #    result = np.concatenate([color, alpha], axis=2)
     result = alpha
     return [result]