from pydantic import BaseModel class ImageBatch(BaseModel): images: list[bytes] filenames: list[str] start_sequence: int end_sequence: int batch_number: int