Update main.py

main.py

@@ -191,14 +191,17 @@ async def process_image(file: UploadFile = File(...), top: int = Form(...), bott
         return {"error": "Model not loaded."}
 
     # Save the uploaded image locally with its original filename
-    with open(file.filename, "wb") as buffer:
+    with open("uploaded_image.jpg", "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
 
+    # 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)
+
     # Process the uploaded image
-    # Load the image and convert it from BGR to RGB
-    frame = cv2.imread(file.filename)
-    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-    aligned_face, instyle, message = model.detect_and_align_image(frame, top, bottom, left, right)
+    aligned_face, instyle, message = model.detect_and_align_image("uploaded_image.jpg", 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