| from aggpt14 import AgGPT14 | |
| from training_data import corpus | |
| if __name__ == "__main__": | |
| model = AgGPT14(corpus, order=4, seed=None) | |
| print("Chat with AgGPT14 (type 'quit' to exit)") | |
| print("-" * 40) | |
| while True: | |
| prompt = input("You: ") | |
| if prompt.lower() in ['quit', 'exit', 'q']: | |
| print("Goodbye!") | |
| break | |
| response = model.ask(prompt, max_tokens=999999, temperature=0.5, top_k=5, text_world_model=False) | |
| print(f"AI: {response}\n") | |