payload_model.py

from pydantic import BaseModel

class PayloadModel(BaseModel):
    """Type check for payload parameters"""
    image: str
    prompt_keyword: str