update

app.py

@@ -1,9 +1,12 @@
 from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import StreamingResponse
 from retinaface import RetinaFace
 import cv2
 import numpy as np
-from typing import Any, Dict
+from typing import Any
 from rembg import remove
+from PIL import Image
+import io
 
 app = FastAPI()
 
@@ -35,12 +38,19 @@ async def detect_faces(file: UploadFile = File(...)):
 @app.post("/remove_background")
 async def remove_background(file: UploadFile = File(...)):
     contents = await file.read()
-    nparr = np.frombuffer(contents, np.uint8)
-    image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
-    # Remove background
-    image = remove(image)
+    image = Image.open(io.BytesIO(contents))
+    
+    image_rembg = remove(image)
+    image_np = np.array(image_rembg)
+    image = Image.fromarray(image_np)
     
-    return image
+    # Convert the image to bytes
+    img_byte_arr = io.BytesIO()
+    image.save(img_byte_arr, format='PNG')
+    img_byte_arr.seek(0)
+    
+    return StreamingResponse(io.BytesIO(img_byte_arr.read()), media_type="image/png")
+
 
 
 @app.get("/")