Update main.py

main.py

@@ -219,7 +219,11 @@ async def process_image(file: UploadFile = File(...), top: int = Form(...), bott
 
     # Convert BGR to RGB
     aligned_face_rgb = cv2.cvtColor(aligned_face, cv2.COLOR_BGR2RGB)
-    instyle_rgb = cv2.cvtColor(instyle, 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')
@@ -236,3 +240,4 @@ app.mount("/", StaticFiles(directory="AB", html=True), name="static")
 @app.get("/")
 def index() -> FileResponse:
     return FileResponse(path="/app/AB/index.html", media_type="text/html")
+