from pydantic import BaseModel class Prediction(BaseModel): message : str = "Enter a text message" message_1 : str = "Enter a text message"