Update app.py

app.py

@@ -2,22 +2,21 @@ import gradio as gr
 from PIL import Image
 import rembg
 import io
-import tempfile
-import os
 
 def remove_background(image):
     # Process image
     img_byte_arr = io.BytesIO()
     image.save(img_byte_arr, format='PNG')
-    result_bytes = rembg.remove(img_byte_arr.getvalue())
-    processed_img = Image.open(io.BytesIO(result_bytes)).convert("RGBA")
+    processed_bytes = rembg.remove(img_byte_arr.getvalue())
     
-    # Save to temporary file for download
-    temp_dir = tempfile.mkdtemp()
-    file_path = os.path.join(temp_dir, "no_background.png")
-    processed_img.save(file_path)
+    # Create PIL image for preview
+    processed_img = Image.open(io.BytesIO(processed_bytes))
     
-    return processed_img, file_path
+    # Prepare download file
+    download_bytes = io.BytesIO(processed_bytes)
+    download_bytes.seek(0)  # Reset buffer position
+    
+    return processed_img, download_bytes
 
 # Gradio Interface
 image_upload = gr.Image(label="Upload Image", type="pil")
@@ -30,6 +29,7 @@ interface = gr.Interface(
     outputs=[output_preview, output_file],
     title=" PNG Background Remover",
     description="Remove background and preview/download as transparent PNG",
+    allow_flagging="never"
 )
 
 if __name__ == "__main__":