11 lines
171 B
Python
11 lines
171 B
Python
from pydantic import BaseModel
|
|
|
|
|
|
class GenerateImageModel(BaseModel):
|
|
tasks_id: str
|
|
prompt: str
|
|
image_url: str
|
|
mode: str
|
|
category: str
|
|
gender: str
|