Update main.py

main.py

@@ -190,13 +190,13 @@ async def process_image(file: UploadFile = File(...), top: int = Form(...), bott
     if model is None:
         return {"error": "Model not loaded."}
 
-    # Save the uploaded image locally
-    with open("uploaded_image.jpg", "wb") as buffer:
+    # Save the uploaded image locally with its original filename
+    with open(file.filename, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
 
     # Process the uploaded image
     # Load the image and convert it from BGR to RGB
-    frame = cv2.imread("uploaded_image.jpg")
+    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)
     processed_image, message = model.image_toonify(aligned_face, instyle, model.exstyle, style_degree=0.5, style_type='cartoon1')