File size: 523 Bytes
194b4ef |
1 2 3 4 5 6 7 8 9 10 11 12 |
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") |