17 lines
379 B
Python
17 lines
379 B
Python
|
from typing import Optional
|
||
|
|
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
|
||
|
|
|
||
|
|
class ClassificationResponse(BaseModel):
|
||
|
|
is_ai_generated: bool
|
||
|
|
ai_probability: float
|
||
|
|
human_probability: float
|
||
|
|
|
||
|
|
|
||
|
|
class IconRequest(BaseModel):
|
||
|
|
prompt: str = Field(..., min_length=1, max_length=4000)
|
||
|
|
color: Optional[str] = None
|
||
|
|
shape: Optional[str] = None
|
||
|
|
style: Optional[str] = None
|