Update main.py

main.py

@@ -18,13 +18,16 @@ class ImageRequest(BaseModel):
     image_file: UploadFile = File(...)
 
 @app.post("/upload/")
-async def toonify_image(image_request: ImageRequest):
-    image = await image_request.image_file.read()
-    nparr = np.frombuffer(image, np.uint8)
-    img = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
-    aligned_face, instyle, message = model.detect_and_align_image(img, 200, 200, 200, 200)  # Hardcoded values
-    toonified_img, message = model.image_toonify(aligned_face, instyle, exstyle, style_degree=0.5, style_type="cartoon1")
-    return {"toonified_image": toonified_img, "message": message}
+async def toonify_image(image_file: UploadFile = File(...)):
+    try:
+        contents = await image_file.read()
+        nparr = np.frombuffer(contents, np.uint8)
+        img = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
+        aligned_face, instyle, message = model.detect_and_align_image(img, 200, 200, 200, 200)  # Hardcoded values
+        toonified_img, message = model.image_toonify(aligned_face, instyle, exstyle, style_degree=0.5, style_type="cartoon1")
+        return {"toonified_image": toonified_img, "message": message}
+    except Exception as e:
+        return {"error": str(e)}
 
 app.mount("/", StaticFiles(directory="AB", html=True), name="static")