10 lines
286 B
Python
10 lines
286 B
Python
|
# app/models/schemas.py
|
||
|
|
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
|
||
|
|
class TextInput(BaseModel):
|
||
|
|
text: str = Field(..., example="Это просто пиииииииииииииииииииздец")
|
||
|
|
|
||
|
|
class ToxicityOutput(BaseModel):
|
||
|
|
toxicity_score: float = Field(..., example=0.85)
|