src/api/schemas.py

from pydantic import BaseModel, Field


class PredictionRequest(BaseModel):
    text: str = Field(..., min_length=5)


class PredictionResponse(BaseModel):
    predicted_class: str
    confidence_score: float