File size: 195 Bytes
16dca74
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
from pydantic import BaseModel

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

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