Update app.py

app.py

@@ -1,22 +1,32 @@
 import gradio as gr
 from rembg import remove
 from PIL import Image
+import base64
 
 def remove_background(input_image):
     input = Image.open(input_image) # load image
     output = remove(input) # remove background
     return output
 
-input_image = gr.inputs.Image(type="filepath", label="Input Image")
-output_image = gr.inputs.Image(type="filepath", label="Output Image")
-output_image_download = gr.outputs.Download(label="Download Output")
+input_image = gr.inputs.Image(type="pil", label="Input Image")
+output_image = gr.outputs.Image(type="pil", label="Output Image")
 
 # Create a Gradio interface
 iface = gr.Interface(remove_background, 
                      inputs=input_image, 
-                     outputs=[output_image, output_image_download], 
+                     outputs=output_image, 
                      title='Image Background Remover',
-                     description='Upload an image and it will remove the background.')
+                     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
 iface.launch()
+
+# After the interface is launched, you can download the output image as follows:
+output_pil = iface.get_output() # get the output PIL image from the