diff --git a/endpoints/yolomax.get.py b/endpoints/yolomax.get.py index 52e0809..ed4f925 100644 --- a/endpoints/yolomax.get.py +++ b/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!"} \ No newline at end of file +@router.get("/yolomax", response_model=YoloResponseSchema, status_code=status.HTTP_200_OK) +async def get_yolo(): + """Return yolooo3!""" + return {"message": "yolooo3!"} \ No newline at end of file