Update app.py

app.py

@@ -1,34 +1,42 @@
+
+
+
+
 import gradio as gr
 from rembg import remove
 from PIL import Image
-import base64
+from io import BytesIO
 
 def remove_background(input_image):
     input = Image.open(input_image) # load image
     output = remove(input) # remove background
     return output
 
+# define input and output interfaces
+
 input_image = gr.inputs.Image(type="filepath", label="Input Image")
 output_image = gr.outputs.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.')
-
-# Define a function to convert PIL image to base64 string
-def pil_to_base64_string(pil_image):
-    buffered = BytesIO()
-    pil_image.save(buffered, format="PNG")
-    img_str = base64.b64encode(buffered.getvalue()).decode()
-    return img_str
-
-# Launch the interface
+# define function for downloading output image
+def download_output(output_image):
+    output_image.save("output.png")
+    with open("output.png", "rb") as f:
+        img_bytes = BytesIO(f.read())
+    return img_bytes
+
+# define output download button
+output_image_download = gr.outputs.Button(label="Download Output", type="bytes", callback=download_output)
+
+# create the interface
+iface = gr.Interface(
+    fn=remove_background, 
+    inputs=input_image, 
+    outputs=output_image,
+    output_options=[output_image_download],
+    title='Image Background Remover',
+    description='Upload an image and it will remove.'
+)
+
+# launch the interface
 iface.launch()
 
-output_pil = iface.get_output() # get the output PIL image from the interface
-output_base64_str = pil_to_base64_string(output_pil) # convert the PIL image to base64 string
-with open("output_image.png", "wb") as f:
-    f.write(base64.b64decode(output_base64_str)) # write the decoded base64 string to a file