from pydantic import BaseModel


class PredictionResult(BaseModel):
    label: str
    confidence: float


class PredictionResponse(BaseModel):
    success: bool
    prediction: PredictionResult
    model: str
    processing_time_ms: float