File size: 195 Bytes
083d486
 
 
 
 
 
 
d79a053
 
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: list[str]
    question: str