Update app.py

app.py

@@ -53,9 +53,10 @@ def process(image):
     # paste the mask on the original image
     new_im = Image.new("RGBA", pil_im.size, (0,0,0,0))
     new_im.paste(orig_image, mask=pil_im)
+    new_orig_image = new_orig_image.convert('RGBA')
 
     # return new_im
-    return [new_im, new_im]
+    return [new_orig_image, new_im]
 
 
 # block = gr.Blocks().queue()