File size: 400 Bytes
c4d5641 308bca9 c4d5641 308bca9 c4d5641 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from dataclasses import dataclass
from pydantic import BaseModel
@dataclass
class Word:
tokens: list[int]
text: str
logprob: float
context: list[int]
class Replacement(BaseModel):
text: str
logprob: float
class ApiWord(BaseModel):
text: str
logprob: float
replacements: list[Replacement]
class CheckResponse(BaseModel):
text: str
words: list[ApiWord]
|