Update app.py

app.py

@@ -1,22 +1,31 @@
 import gradio as gr
 from rembg import remove
 from PIL import Image
+import io
 
 def remove_background(input_image):
-
     input = Image.open(input_image) # load image
     output = remove(input) # remove background
     return output
 
+def download_output(output_image):
+    output_bytes = io.BytesIO()
+    output_image.save(output_bytes, format="PNG")
+    output_bytes.seek(0)
+    return output_bytes
+
+input_image = gr.inputs.Image(type="file", label="Input Image")
+output_image = gr.outputs.Image(type="pil", label="Output Image")
+output_image_download = gr.outputs.Button(label="Download Output")
 
-input_image = gr.inputs.Image(type="filepath", label="Input Image")
-output_image = gr.inputs.Image(type="filepath", label="Output Image")
-# Create a Gradio interface
-iface = gr.Interface(remove_background, 
-                     inputs=input_image, 
-                     outputs=output_image, 
-                     title='Image Background Remover',
-                     description='Upload an image and it will remove.')
+def background_removal_with_download(input_image):
+    output_image = remove_background(input_image)
+    output_image_download = gr.outputs.Button(label="Download Output")
+    download_button = gr.Interface(
+        download_output, inputs=gr.inputs.Image(type="pil", label="Output Image"), outputs="bytes",
+        interpretation="download", title="Download Output Image", live=False)
+    return output_image, download_button(output_image)
 
-# Launch the interface
+iface = gr.Interface(background_removal_with_download, inputs=input_image, outputs=[output_image, output_image_download],
+                     title='Image Background Remover', description='Upload an image and it will remove.')
 iface.launch()