Skip to content

Latest commit

 

History

History
17 lines (10 loc) · 1.01 KB

README.md

File metadata and controls

17 lines (10 loc) · 1.01 KB

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