from pydantic import BaseModel


class GenerateImageModel(BaseModel):
    category: str
    content: str
    gender: str
    image_url: str
    mode: int
    tasks_id: str
    user_id: int
    version: str