from pydantic import BaseModel class PredictRequest(BaseModel): distance_km: float class PredictResponse(BaseModel): cost_soles: float