from pydantic import BaseModel

class SingleInferencePayload(BaseModel):
    image_path: str
    question: str

class VideoInferencePayload(BaseModel):
    video_path: list[str]
    question: str