backend/models.py

from pydantic import BaseModel
from typing import List


class ImageInpaintingRequest(BaseModel):
    images: List[str]
    alpha_gradient_width: int
    init_image_height: int
    
    class Config:
        arbitrary_types_allowed = True

class ImageInpaintingResponse(BaseModel):
    inpainted_image: str

    class Config:
        arbitrary_types_allowed = True