from pydantic import BaseModel


class PredictionRequest(BaseModel):
    question: str


class PredictionResponse(BaseModel):
    response: str