payload_model.py

from pydantic import BaseModel

class PayloadModel(BaseModel):
    """Type check for payload parameters"""
    input_data: list[str]
    prompt: str
    prompt_eval_mode: str
    input_utilization_mode: str
    bbox: list[list[list[int]]]