Update main.py

main.py

@@ -54,7 +54,11 @@ async def cartoonize_image(file: UploadFile = File(...)):
     image = Image.open(io.BytesIO(contents))
     image = np.array(image)
     cartoonized_image = inference(image)
-    return StreamingResponse(io.BytesIO(cv2.imencode('.jpg', cartoonized_image)[1]), media_type="image/jpeg")
+    
+    # Convert BGR to RGB format
+    cartoonized_image_rgb = cv2.cvtColor(cartoonized_image, cv2.COLOR_BGR2RGB)
+    
+    return StreamingResponse(io.BytesIO(cv2.imencode('.jpg', cartoonized_image_rgb)[1]), media_type="image/jpeg")
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")