adjust mnist url

app/main.py

@@ -5,7 +5,7 @@
 from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 
-from app.routers import base_router, predict_router
+from app.routers import base_router, mnist_router
 from app.vision.mnist import init_vision_model
 
 
@@ -31,7 +31,7 @@ async def lifespan(app: FastAPI):
 )
 mnist_app.mount("/static", StaticFiles(directory="static"), name="static")
 mnist_app.include_router(base_router)
-mnist_app.include_router(predict_router)
+mnist_app.include_router(mnist_router)
 
 
 async def init_model() -> None:

app/routers/__init__.py

@@ -1,5 +1,5 @@
 from .base import router as base_router
-from .predict import router as predict_router
+from .mnist import router as mnist_router
 
 BASE_ROUTERS = [base_router]
-PREDICT_ROUTERS = [predict_router]
+PREDICT_ROUTERS = [mnist_router]

app/routers/base.py

@@ -10,7 +10,7 @@
     mnist_content = file.read()
 
 
-@router.get("/", response_class=HTMLResponse)
+@router.get("/mnist", response_class=HTMLResponse)
 async def read_root() -> HTMLResponse:
     """
     Serve the MNIST HTML page from the static directory.

tests/app/test_base.py

@@ -7,7 +7,7 @@ class TestBase:
 
     @pytest.mark.asyncio
     def test_get_root(self, client: TestClient) -> None:
-        response = client.get(f"/")
+        response = client.get(f"/mnist")
 
         assert response.status_code == status.HTTP_200_OK
         assert response.headers["Content-Type"] == "text/html; charset=utf-8"

tests/app/test_predict.py

@@ -7,7 +7,7 @@
 
 class TestPredictDigit:
 
-    @patch("app.routers.predict.predict_digit")
+    @patch("app.routers.mnist.predict_digit")
     def test_predict_digit(
         self, mock: MagicMock, client: TestClient, create_mnist_image: tuple[str, int]
     ) -> None: