from pydantic import BaseModel class FaceData(BaseModel): image: str target_image: str def to_dict(self): return { 'image': self.image, 'target_image': self.target_image, }