Update main.py

main.py

@@ -216,17 +216,9 @@ async def process_image(file: UploadFile = File(...), top: int = Form(...), bott
 
     # Process the uploaded image
     aligned_face, instyle, message = model.detect_and_align_image("uploaded_image.jpg", top, bottom, left, right)
+    instyle_dict = {'instyle': instyle.numpy()}  # Convert instyle to a dictionary containing a NumPy array
 
-    # Convert BGR to RGB
-    aligned_face_rgb = cv2.cvtColor(aligned_face, cv2.COLOR_BGR2RGB)
-    # Convert instyle tensor to numpy array
-    instyle_np = instyle.cpu().numpy()
-
-    # Convert BGR to RGB
-    instyle_rgb = cv2.cvtColor(instyle_np, cv2.COLOR_BGR2RGB)
-
-    # Toonify the image
-    processed_image, message = model.image_toonify(aligned_face_rgb, instyle_rgb, model.exstyle, style_degree=0.5, style_type='cartoon3-d')
+    processed_image, message = model.image_toonify(aligned_face, instyle_dict, model.exstyle, style_degree=0.5, style_type='cartoon3-d')
 
     # Convert processed image to bytes
     image_bytes = cv2.imencode('.jpg', processed_image)[1].tobytes()