File size: 619 Bytes
f2878d0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | """TinyDecide: ask many typed questions about one message, get calibrated probabilities in one pass.
from tinydecide import TinyDecide
model = TinyDecide.load("path/to/TinyDecide") # or TinyDecide.from_pretrained()
r = model.answer("Book a table for 4 on Friday at 7:30",
[{"type": "span", "text": "Extract the time."}])
r["answers"][0]["text"] # "7:30"
"""
from .corrections import make_protos
from .engine import TinyDecide, WordPiece, encode_request
__all__ = ["TinyDecide", "WordPiece", "encode_request", "make_protos"]
__version__ = "0.1.0"
|