Spaces:

mebubo
/

gpted

Runtime error

gpted / models.py

Working probs in UI

308bca9 almost 2 years ago

400 Bytes

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]

	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]