/gpt2_chatbot

☕ GPT-2 chatbot for daily conversation

Primary LanguagePythonMIT LicenseMIT

☕ GPT2 Chatbot

GPT-2 chatbot for daily conversations trained on Daily Dialogue, Empathetic Dialogues, PERSONA-CHAT, Blended Skill Talk datasets. This chatbot is made based on GPT2 Model transformer with a language modeling head on top.

chatbot

⌛ Installation

Download the model from AWS S3 storage and run the following command:

git pull https://github.com/xcapt0/gpt2_chatbot.git
docker build -t gpt2_bot .

🤖 Usage

Run the docker container:

docker run --rm -it gpt2_bot

There are 2 different ways to use the chatbot: train and interact mode

Interaction mode

To launch the chatbot run the following command. Specify --checkpoint path to your model

python chatbot.py --mode interact --checkpoint path/to/model.h5

Train mode

To train the model run the following command. Specify --checkpoint if you needed

python chatbot.py --mode train

📝 License

Copyright © 2022 xcapt0.
This project is MIT licensed.