Update app.py

app.py

@@ -4,16 +4,17 @@ import rembg
 import io
 
 def remove_background(image):
-    # Process image to bytes
+    # Convert image to bytes
     img_byte_arr = io.BytesIO()
     image.save(img_byte_arr, format='PNG')
     processed_bytes = rembg.remove(img_byte_arr.getvalue())
     
-    # Create PIL image for preview
+    # Prepare outputs
     processed_img = Image.open(io.BytesIO(processed_bytes))
+    file_obj = io.BytesIO(processed_bytes)
+    file_obj.name = "no_background.png"  # Required for Gradio
     
-    # Return (preview image, (filename, bytes)) for download
-    return processed_img, ("no_background.png", processed_bytes)
+    return processed_img, file_obj
 
 # Gradio Interface
 image_upload = gr.Image(label="Upload Image", type="pil")