from pydantic import BaseModel class PredictRequest(BaseModel): predict_text: str