23 lines
372 B
Python
23 lines
372 B
Python
from pydantic import BaseModel
|
|
|
|
|
|
class GenerateImageModel(BaseModel):
|
|
tasks_id: str
|
|
prompt: str
|
|
image_url: str
|
|
mode: str
|
|
category: str
|
|
gender: str
|
|
|
|
|
|
class GenerateSingleLogoImageModel(BaseModel):
|
|
tasks_id: str
|
|
prompt: str
|
|
image_url: str
|
|
|
|
|
|
class GenerateProductImageModel(BaseModel):
|
|
tasks_id: str
|
|
prompt: str
|
|
image_url: str
|