Update app.py

app.py

@@ -53,10 +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 = orig_image.convert('RGBA')
+    # new_orig_image = orig_image.convert('RGBA')
 
-    # return new_im
-    return [new_orig_image, new_im]
+    return new_im
+    # return [new_orig_image, new_im]
 
 
 # block = gr.Blocks().queue()
@@ -94,9 +94,9 @@ gr.HTML('''
 title = "Background Removal"
 description = "Remove background from any image"
 examples = [['./input.jpg'],]
-output = ImageSlider(position=0.5,label='Image without background', type="pil", show_download_button=True)
-demo = gr.Interface(fn=process,inputs="image", outputs=output, examples=examples, title=title, description=description)
-# demo = gr.Interface(fn=process,inputs="image", outputs="image", examples=examples, title=title, description=description)
+# output = ImageSlider(position=0.5,label='Image without background', type="pil", show_download_button=True)
+# demo = gr.Interface(fn=process,inputs="image", outputs=output, examples=examples, title=title, description=description)
+demo = gr.Interface(fn=process,inputs="image", outputs="image", examples=examples, title=title, description=description)
 
 if __name__ == "__main__":
     demo.launch(share=False)