Lemorra's picture
Update src/utils/payload_model.py
d79a053 verified
raw
history blame contribute delete
195 Bytes
from pydantic import BaseModel
class SingleInferencePayload(BaseModel):
image_path: str
question: str
class VideoInferencePayload(BaseModel):
video_path: list[str]
question: str