33 lines
555 B
Python
33 lines
555 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
|
|
seed: str
|
|
|
|
|
|
class GenerateProductImageModel(BaseModel):
|
|
tasks_id: str
|
|
prompt: str
|
|
image_url: str
|
|
image_strength: float
|
|
product_type: str
|
|
|
|
|
|
class GenerateRelightImageModel(BaseModel):
|
|
tasks_id: str
|
|
prompt: str
|
|
image_url: str
|
|
direction: str
|
|
product_type: str
|