Training the chatbot model