Discord Chatbot: The World Ends With You

This is a Discord AI Chatbot that uses the Microsoft DialogGPT conversational model fine-tuned on the game transcript of The World Ends With You (TWEWY).

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 project

Resource Links

• My TWEWY dataset on Kaggle
• My Hugging Face Model
• My Repl.it discord_bot.py script