| from pydantic import BaseModel, Field | |
| class SingleImageRequest(BaseModel): | |
| image: str = Field(default="", title="Image", description="Image to work on, must be a Base64 string containing the image's data.") | |
| class SingleImageResponse(BaseModel): | |
| image: str = Field(default="", title="Image", description="Generated image, a Base64 string containing the image's data.") | |
| class GenerateMaskRequest(SingleImageRequest): | |
| mask_type: int = Field(default="", title="Mask Type", description="Mask type to work with") |