https://github.com/xcapt0/gpt2_chatbot
☕ GPT-2 chatbot for daily conversation
https://github.com/xcapt0/gpt2_chatbot
chatbot gpt-2 nlp
Last synced: 3 months ago
JSON representation
☕ GPT-2 chatbot for daily conversation
- Host: GitHub
- URL: https://github.com/xcapt0/gpt2_chatbot
- Owner: xcapt0
- License: mit
- Created: 2021-12-29T10:56:49.000Z (over 4 years ago)
- Default Branch: master
- Last Pushed: 2022-02-14T09:31:27.000Z (over 4 years ago)
- Last Synced: 2023-08-31T01:33:25.541Z (almost 3 years ago)
- Topics: chatbot, gpt-2, nlp
- Language: Python
- Homepage:
- Size: 22.1 MB
- Stars: 13
- Watchers: 1
- Forks: 1
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# ☕ 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.
## ⌛ Installation
Download the [model](https://gpt2chatbot.s3.us-east-2.amazonaws.com/model.h5) from AWS S3 storage and run the following command:
```sh
git pull https://github.com/xcapt0/gpt2_chatbot.git
docker build -t gpt2_bot .
```
## 🤖 Usage
Run the docker container:
```sh
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
```sh
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
```sh
python chatbot.py --mode train
```
## 📝 License
Copyright © 2022 [xcapt0](https://github.com/xcapt0).
This project is [MIT](https://github.com/xcapt0/gpt2_chatbot/blob/master/LICENSE) licensed.