Update main.py

main.py

@@ -30,25 +30,22 @@ async def process_image(file: UploadFile = File(...), top: int = Form(...), bott
     if model is None:
         load_model()
 
-    # Save the uploaded image locally with its original filename
-    with open("uploaded_image.jpg", "wb") as buffer:
-        shutil.copyfileobj(file.file, buffer)
+    # Read the uploaded image file
+    contents = await file.read()
 
-    # Read the saved image using OpenCV
-    frame = cv2.imread("uploaded_image.jpg")
-
-    # Convert the image from BGR to RGB
-    frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+    # Convert the uploaded image to numpy array
+    nparr = np.frombuffer(contents, np.uint8)
+    frame_rgb = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
 
     # Process the uploaded image
     aligned_face, instyle, message = model.detect_and_align_image(frame_rgb, top, bottom, left, right)
     processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='cartoon1')
 
     # Convert processed image to bytes
-    image_bytes = cv2.imencode('.jpg', processed_image)[1].tobytes()
+    _, encoded_image = cv2.imencode('.jpg', processed_image)
 
     # Return the processed image as a streaming response
-    return StreamingResponse(BytesIO(image_bytes), media_type="image/jpeg")
+    return StreamingResponse(io.BytesIO(encoded_image.tobytes()), media_type="image/jpeg")
 
 # Mount static files directory
 app.mount("/", StaticFiles(directory="AB", html=True), name="static")