An AI discord bot that can simulate a conversation with RFA's best hacker, 707 from Mystic Messenger!
Credits to Lynn Zheng and her Discord AI Chatbot tutorial as well as Rostyslav Neskorozhenyi and his post about making a Rick Sanchez bot for helping me build this.
The chatbot uses the Microsoft DialoGPT conversational model that has been trained with 707's messages in the Deep Story from this Kaggle dataset. The dataset has almost 4000 lines and I trained it for 12 epochs to give a perplexity of around 1.3. The perplexity represents how confused the model is so the higher the perplexity, the more confused the model is. 1.3 is a very low perplexity but the chat bot still has some out-of-character responses. With this amount of data, the model took around 1.5 hours to train.
If you want to try out my chatbot, you can go here where it's hosted on Hugging Face's Model Hub.
model_train_upload_workflow.ipynb: Notebook to be run in Google Colab to train and upload the model to Hugging Face's Model Hubmain.py: Script to be imported into a Repl.it Python Discord.py projectkeep_alive.py: Script to keep the discord bot alive for half an hour after main.py is run