Skip to content

shz-code/chatbot

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pretrained NPL model based chatbot using Python

Packages used

  • Speech Recognition
  • PyTorch
  • Pyttsx3
  • Transformers

Project requirements

  • Python 3.7-3.9 (Pytorch limitation)

Installation Process

First clone this git repository or download zip

  git clone https://github.com/shz-code/chatbot.git

Create a new virtual environment(Use Conda/ Virtual Environment) Learn More.

*If virtual environment is not installed on your machine install it using below command.

  pip install virtualenv

Activate virtualenv

   virtualenv env
  .\env\Scripts\activate  

Run pip to install all the dependencies

  pip install -r requirements.txt

Start main.py

  py main.py

Features

  • Users can chat or speak with the bot.
  • The bot can generate answers based on pre-defined conditions or it will generate answer from NLP model.
  • microsoft/DialoGPT-medium used as pre-trained model. Change the model name and task (in this project conversational) to add other pre-trained models. Visit Hugging Face to know more.

Acknowledgements