20 lines
319 B
Python
20 lines
319 B
Python
from pydantic import BaseModel
|
|
|
|
|
|
class GenerateImageModel(BaseModel):
|
|
tasks_id: str
|
|
prompt: str
|
|
image_url: str
|
|
|
|
|
|
class GenerateSingleLogoImageModel(BaseModel):
|
|
tasks_id: str
|
|
prompt: str
|
|
seed: str
|
|
|
|
|
|
class GenerateProductImageModel(BaseModel):
|
|
tasks_id: str
|
|
prompt: str
|
|
image_url: str
|