api/schema.py

from pydantic import BaseModel


class ClaimRequest(BaseModel):
    claim: str


class PredictionResponse(BaseModel):
    prediction: str