Update app.py

app.py

@@ -235,14 +235,15 @@ def inpaint(image, invert):
     else:
 #        alpha = 255 - image["mask"][:, :, 0:1]
         alpha = 255 - image["layers"][0][:, :, 0:1]
-    result = np.concatenate([color, alpha], axis=2)
+#    result = np.concatenate([color, alpha], axis=2)
+    result = alpha
     return [result]
 
 
 def predict(im):
 #    return im["composite"]
 #    return im["inputmask"] # bad
-    return im["layers"][0]
+    return im["layers"][0][:, :, 0:1]
 
 block = gr.Blocks().queue()
 with block: