TinyDecide / python /tinydecide /__init__.py
TheREZOR's picture
Python, Rust and ESP32 engines, shared conformance set, promo video
f2878d0 verified
Raw History Blame Contribute Delete
619 Bytes
"""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"