Update code in endpoints/yolomax.get.py

endpoints/yolomax.get.py

@@ -1,10 +1,11 @@
-# Entity: YoloMessage
+# Entity: YoloResponse
 
 from fastapi import APIRouter, status
+from schemas.yolo import YoloResponseSchema
 
 router = APIRouter()
 
-@router.get("/yolomax", status_code=status.HTTP_200_OK)
-async def get_yolo_message():
-    """Return yolooo2! message"""
-    return {"message": "yolooo2!"}
+@router.get("/yolomax", response_model=YoloResponseSchema, status_code=status.HTTP_200_OK)
+async def get_yolo():
+    """Return yolooo3!"""
+    return {"message": "yolooo3!"}