/DiscordChatAI-GPT2

A chat AI discord bot written in python3 using GPT-2, trained on data scraped from every message of my discord server (can be trained on yours too)

Primary LanguagePython

discord gpt2 chatbot

(based on https://github.com/NickBrisebois/DiscordChatAI-GPT2) uses Python 3

Bot responds to mentions and also has a 25% chance of responding to random messages. Bot can also briefly remember previous messages and use their context.

To train:

Run python3 main.py --genmodel to train. Requires a .txt file of the text channel history to be placed in the same directory as name called "chat.txt". An example of the chat.txt format is provided. It's defaulted to the 355M dataset, but you can change its value in main.py and in Bot/ai.py.

To test:

You can test this bot locally by running python3 main.py -t. It will let you input some text and it will print the AI's response.

To run on Discord:

python3 main.py --token [discord bot token]