schema.py

from pydantic import BaseModel

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