Update app.py

app.py

@@ -4,23 +4,18 @@ from PIL import Image
 import io
 
 def background_remover(input_image):
-    output_img = remove(input_image)  # Directly use the PIL Image object
-    
-    buf = io.BytesIO()
-    output_img.save(buf, format='PNG')
-    buf.seek(0)
-    return buf
-
+    output_img = remove(input_image)  # Remove the background using the rembg library
+    # No need to convert to BytesIO, just return the PIL Image object
+    return output_img
 
 description = """
 This app uses the rembg library to remove the background from uploaded images.
 Simply upload your image and let the model do its work. Download the result immediately after!
 """
 
-# Create the Gradio interface
 iface = gr.Interface(
     fn=background_remover,
-    inputs=gr.Image(type='pil', label="Upload Image"),
+    inputs=gr.inputs.Image(type='pil', label="Upload Image"),
     outputs="image",
     examples=["woman.jpg", "groot.jpg"],
     title="Background Remover",