from fastai.text.all import load_learner

def chat_with_model():
    learn = load_learner('model.pkl')
    context = "Hey"
    while True:
        print(f"You: {context}")
        generated = learn.predict(context, 200, temperature=0.9)
        print(f"Bot: {generated}\n")

        user_input = input("You: ")
        if user_input.strip().lower() in ["quit", "exit", "stop"]:
            break
        context = user_input

if __name__ == "__main__":
    chat_with_model()