Update app.py

app.py

@@ -4,7 +4,7 @@ import rembg
 import io
 
 def remove_background(image):
-    # Process image
+    # Process image to bytes
     img_byte_arr = io.BytesIO()
     image.save(img_byte_arr, format='PNG')
     processed_bytes = rembg.remove(img_byte_arr.getvalue())
@@ -12,11 +12,8 @@ def remove_background(image):
     # Create PIL image for preview
     processed_img = Image.open(io.BytesIO(processed_bytes))
     
-    # Prepare download file
-    download_bytes = io.BytesIO(processed_bytes)
-    download_bytes.seek(0)  # Reset buffer position
-    
-    return processed_img, download_bytes
+    # Return (preview image, (filename, bytes)) for download
+    return processed_img, ("no_background.png", processed_bytes)
 
 # Gradio Interface
 image_upload = gr.Image(label="Upload Image", type="pil")