fxrcode / DialoGPT-chatbot

forked from RuolinZheng08/twewy-discord-chatbot

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

forked from RuolinZheng08/twewy-discord-chatbot

License:MIT License


Languages

Language:Jupyter Notebook 98.5%Language:Python 0.9%Language:JavaScript 0.5%