Skip to content

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

License

Notifications You must be signed in to change notification settings

bryankhan/twewy-discord-chatbot

 
 

Repository files navigation

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

About

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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 95.0%
  • Python 3.2%
  • JavaScript 1.8%