Upload 2 files

models/requests.py

@@ -0,0 +1,12 @@
+from typing import Dict
+from pydantic import BaseModel
+
+
+class PredictRequest(BaseModel):
+    """Request model for /predict endpoint.
+
+    Contains a mapping of feature names to numeric values that the model
+    will consume. Keep it minimal so imports from `models.requests`
+    succeed when the app starts.
+    """
+    features: Dict[str, float]

models/responses.py

@@ -0,0 +1,11 @@
+from typing import Dict, List, Optional
+from pydantic import BaseModel
+
+class PredictResponse(BaseModel):
+    label: str
+    probabilities: Optional[Dict[str, float]] = None
+    recognized: List[str]
+    unknown: List[str]
+    missing: List[str]
+    feature_order: List[str]
+    engineered: Dict[str, Optional[float]]