Update main.py

main.py

@@ -21,11 +21,12 @@ class ImageRequest(BaseModel):
 async def toonify_image(image_file: UploadFile = File(...)):
     try:
         contents = await image_file.read()
+        filename = image_file.filename
         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}
+        return {"toonified_image": toonified_img, "message": message, "filename": filename}
     except Exception as e:
         return {"error": str(e)}