Harry Potter Discord AI ChatBot

This is a Discord AI Chatbot that uses the Microsoft DialoGPT conversational model fine-tuned on the book transcript of Harry Potter (HP).

I trained the model using the lines of my favorite character, Harry Potter. He has about 155 lines in total in the transcript I used.

You can 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 replit.com Python Discord.py project

Resource Links

• My Harry Potter dataset on Kaggle
• My Hugging Face Model