| from __future__ import annotations | |
| from model import Add2ModelError, TinyAdd2LLM | |
| def main() -> None: | |
| model = TinyAdd2LLM.load("model/add2_model.bin") | |
| print("Tiny Add2 LLM loaded.") | |
| print(f"Limit: 0 to {model.max_number}. Type 'exit' to stop.") | |
| print("Examples: 12 + 30 | add 15 and 7 | 100 plus 55") | |
| while True: | |
| user = input("You: ").strip() | |
| if user.lower() in {"exit", "quit", "q"}: | |
| print("Bye") | |
| break | |
| try: | |
| print("TinyAdd2LLM:", model.answer(user)) | |
| except Add2ModelError as e: | |
| print("TinyAdd2LLM:", e) | |
| if __name__ == "__main__": | |
| main() | |