gpt2_chatbot

☕ Chatbot GPT2

Chatbot GPT-2 para conversas diárias treinadas em conjuntos de dados Daily Dialogue , Empathetic Dialogues , PERSONA-CHAT , Blended Skill Talk . Este chatbot é feito com base no transformador modelo GPT2 com um cabeçote de modelagem de linguagem na parte superior.

⌛ Instalação

Baixe o modelo do armazenamento AWS S3 e execute o seguinte comando:

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

? Uso

Execute o contêiner do Docker:

docker run --rm -it gpt2_bot

Existem 2 maneiras diferentes de usar o chatbot: modo train e interact

Modo de interação

Para iniciar o chatbot execute o seguinte comando. Especifique o caminho --checkpoint para o seu modelo

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

Modo trem

Para treinar o modelo, execute o seguinte comando. Especifique --checkpoint se precisar

python chatbot.py --mode train

Licença

Direitos autorais © 2022 xcapt0.
Este projeto é licenciado pelo MIT.