Update app.py

app.py

@@ -228,12 +228,15 @@ model_inpaint = None
 def inpaint(image, invert):
 #    image = resize_image(img, res)
 #    color = HWC3(image["image"])
-#    if(invert):
+    color = HWC3(image["composite"])
+    if(invert):
 #        alpha = image["mask"][:, :, 0:1]
-#    else:
+        alpha = image["background"][:, :, 0:1]
+    else:
 #        alpha = 255 - image["mask"][:, :, 0:1]
+        alpha = 255 - image["background"][:, :, 0:1]
 #    result = np.concatenate([color, alpha], axis=2)
-    result = image["mask"]
+    result = np.concatenate(color)
     return [result]
 
 block = gr.Blocks().queue()
@@ -262,7 +265,6 @@ with block:
     with gr.Row():
         with gr.Column():
             input_image = gr.ImageMask(sources="upload", type="numpy", height="auto")
-            input_preview=gr.Image()
 #            input_image = gr.ImageEditor(sources="upload", type="numpy", height="auto", layers="False", brush=gr.Brush(colors=["#000000"]))
 #            input_image = gr.Image(source='upload', type="numpy", tool="sketch", height=512)
 #            resolution = gr.Slider(label="resolution", minimum=256, maximum=1024, value=512, step=64)