great-ai/example/models/domain_prediction.py
András Schmelczer d79a75d352 Add files
Signed-off-by: András Schmelczer <andras@schmelczer.dev>
2022-04-02 13:47:33 +02:00

8 lines
No EOL
157 B
Python

from pydantic import BaseModel
from typing import List
class DomainPrediction(BaseModel):
domain: str
probability: float
explanation: List[str]