diff --git a/apps/ml-service/schemas.py b/apps/ml-service/schemas.py new file mode 100644 index 0000000..4ef3ccd --- /dev/null +++ b/apps/ml-service/schemas.py @@ -0,0 +1,21 @@ +from pydantic import BaseModel + + +class HealthResponse(BaseModel): + status: str + model_loaded: bool + + +class MetadataResponse(BaseModel): + service_name: str + service_version: str + model_path: str + model_loaded: bool + input_size: int + labels: list[str] + + +class PredictionResponse(BaseModel): + label: str + confidence: float + probabilities: dict[str, float]