/DeepChat

Chatbot framework using huggingface

Primary LanguagePythonApache License 2.0Apache-2.0

License

DeepChat

Chatbot framework using Huggingface transformers

This is meant to be a wrapper around huggingface, or potentially other frameworks and models to make chatbot development easier and fun to play around with.

Supported Models

Usage

TBD

Examples

How to start a simple chatbot in a few lines of code using one of the pretrained Huggingface models.

from deepchat import DeepChat

# create DeepChat object that uses small model and default settings
dc = DeepChat("dialogpt", device_type="cuda")

# begins the chatbot conversation
dc.run()