Skip to content

Commit cad59a5

Browse files
committed
2 parents eca0599 + 8901013 commit cad59a5

1 file changed

Lines changed: 14 additions & 3 deletions

File tree

README.md

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,25 @@
22

33
This is a Discord AI Chatbot that uses the [Microsoft DialogGPT conversational model](https://huggingface.co/microsoft/DialoGPT-medium) fine-tuned on the game transcript of [The World Ends With You](https://en.wikipedia.org/wiki/The_World_Ends_with_You) (TWEWY).
44

5-
I trained the model using the lines of my favorite quirky character, Joshua. He has about 700 lines in total in the entire game.
5+
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.
66

77
<img src="https://github.com/RuolinZheng08/twewy-discord-chatbot/blob/main/gif-demo/gameplay.png" width=400><br>
88

99
Here is a demo of the Discord bot in action.
1010

11-
<img src="https://github.com/RuolinZheng08/twewy-discord-chatbot/blob/main/gif-demo/discord.gif" width=600><br>
11+
<img src="https://github.com/RuolinZheng08/twewy-discord-chatbot/blob/main/gif-demo/discord.gif" width=500><br>
1212

1313
You can also directly chat with the model hosted on [Hugging Face's Model Hub](https://huggingface.co/r3dhummingbird/DialoGPT-medium-joshua).
1414

15-
<img src="https://github.com/RuolinZheng08/twewy-discord-chatbot/blob/main/gif-demo/huggingface.gif" width=600><br>
15+
<img src="https://github.com/RuolinZheng08/twewy-discord-chatbot/blob/main/gif-demo/huggingface.gif" width=400><br>
16+
17+
## Structure of this Project
18+
19+
- `model_train_upload_workflow.ipyb`: Notebook to be run in Google Colab to train and upload the model to Hugging Face's Model Hub
20+
- `discord_bot.py`: Script to be imported into a Repl.it project
21+
22+
## Resource Links
23+
24+
- [My TWEWY dataset on Kaggle](https://www.kaggle.com/ruolinzheng/twewy-game-script)
25+
- [My Hugging Face Model](https://huggingface.co/r3dhummingbird/DialoGPT-medium-joshua)
26+
- [My Repl.it `discord_bot.py` script](https://replit.com/@r3dhummingbird/TWEWY-Discord-Chatbots-HuggingFace#main.py)

0 commit comments

Comments
 (0)