Update app.py

app.py

@@ -1,22 +1,13 @@
-from fastapi import FastAPI, File, UploadFile
+# app.py
+from fastapi import FastAPI, UploadFile, File
 from fastapi.responses import StreamingResponse
 from rembg import remove
-from PIL import Image
 import io
 
 app = FastAPI()
 
 @app.post("/remove-bg")
 async def remove_bg(file: UploadFile = File(...)):
-    input_bytes = await file.read()
-    input_image = Image.open(io.BytesIO(input_bytes)).convert("RGBA")
-    
-    # Remove background
+    input_image = await file.read()
     output_image = remove(input_image)
-
-    # Convert output to byte stream
-    img_byte_arr = io.BytesIO()
-    output_image.save(img_byte_arr, format='PNG')
-    img_byte_arr.seek(0)
-
-    return StreamingResponse(img_byte_arr, media_type="image/png")
+    return StreamingResponse(io.BytesIO(output_image), media_type="image/png")