RuolinZheng08 / twewy-discord-chatbot

Discord AI Chatbot using DialoGPT, trained on the game transcript of The World Ends With You

Build a Discord AI Chatbot that Speaks like Your Favorite Character!

This is a Discord AI Chatbot that uses the Microsoft DialoGPT conversational model fine-tuned on the game transcript of The World Ends With You (TWEWY). Read my tutorial on freeCodeCamp or watch my video tutorial on YouTube. I've also made a JavaScript version of the tutorial using Discord.js.

I trained the model using the lines of my favorite quirky character, Joshua (left in the image below). He has about 700 lines in total in the entire game.

Here is a demo of the Discord bot in action.

You can also directly chat with the model hosted on Hugging Face's Model Hub.

Structure of this Project

• model_train_upload_workflow.ipyb: Notebook to be run in Google Colab to train and upload the model to Hugging Face's Model Hub
• discord_bot.py: Script to be imported into a Repl.it Python Discord.py project
• discord_bot.js: Script to be imported into a Repl.it JavaScript Discord.js project

Resource Links

• 15-min chat demo
• My tutorial on freeCodeCamp
• My video tutorial on YouTube
• My JavaScript version of this tutorial on YouTube
• My TWEWY dataset on Kaggle
• My Hugging Face Model