fix(model) : fix model id and fix docs

src/controllers/home_controller.py

@@ -8,7 +8,7 @@ from sqlalchemy.orm import Session
 from config.db import get_db
 from models.ml import MLModel
 
-router = APIRouter(prefix="/models", tags=["Models"])
+router = APIRouter(tags=["Models"])
 
 
 class MLModelOut(BaseModel):

src/controllers/predict_controller.py

@@ -146,7 +146,7 @@ def batch_predict(
 
             db.add(
                 MLOutput(
-                    input_id=str(objs[idx].id),
+                    input_id=objs[idx].id,
                     prediction=label,
                     prob=float(p[i]),
                 )

src/models/ml_output.py

@@ -22,7 +22,7 @@ class MLOutput(Base):
         DateTime(timezone=True), server_default=func.now()
     )
 
-    input_id: Mapped[str] = mapped_column(
+    input_id: Mapped[uuid.UUID] = mapped_column(
         UUID(as_uuid=True), ForeignKey("ml_inputs.id", ondelete="CASCADE"), nullable=False
     )