12 lines
523 B
Python
12 lines
523 B
Python
|
|
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") |