Update app.py

app.py

@@ -228,7 +228,7 @@ model_inpaint = None
 def inpaint(image, invert):
 #    image = resize_image(img, res)
 #    color = HWC3(image["image"])
-#    color = HWC3(image["background"])
+    color = HWC3(image["background"])
 #    if(invert):
 #        alpha = image["mask"][:, :, 0:1]
 #        alpha = image["layers"][0]
@@ -238,13 +238,13 @@ def inpaint(image, invert):
 #    result = np.concatenate([color, alpha], axis=2)
 #    result = alpha
 #    return [result]
-    return [image]
+    return [color]
 
 
 def predict(im):
 #    return im["composite"]
 #    return im["inputmask"] # bad
-    return im["layers"][0][:, :, 0:1]
+    return im["layers"][0][:, :, :]
 
 block = gr.Blocks().queue()
 with block: