Skip to content

Files

Latest commit

 

History

History
52 lines (41 loc) · 1.86 KB

README.md

File metadata and controls

52 lines (41 loc) · 1.86 KB

UniChatBot: Universal chatbot based on DialoGPT

Requirements

  • Python environment 3.6
conda create -n unichatbot_env python=3.6
conda activate unichatbot_env
  • clone the repo and cd into it
git clone https://github.com/Jahysama/UniChatBot
cd UniChatBot
  • python packages
pip install -r requirements.txt

Pre-trained Models

Model Link to the model
English HuggingFace
Russian HuggingFace

Basic usage

python unibot.py --lang en

Firstly it will download a chosen model.

After downloading it will ask you to select the source of input data.

Just right-click the last message in your messenger.

Then left-click on the input panel of your messenger to select the bot output.

Here is a quick demo below for DISCORD:

how-to-use-bot GIF

All available options

Option Description Values
--lang Bot model language en/ru
--memory How many messages should bot remember. int
--max-mess-len Max message length in symbols. int
--temperature The more is temperature the more nonsense bot will generate. float
--response How often bot would check chat and respond in seconds float

License

GNU GPL v3