from pydantic import BaseModel
class GenerateImageModel(BaseModel):
tasks_id: str
prompt: str
image_url: str
mode: str
category: str