/twewy-discord-chatbot

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

Primary LanguageJupyter NotebookMIT LicenseMIT

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