blather

Train and use generative text models in a few lines of code.

To see blather in action check out the colab notebook!

Installation

Use the package manager pip to install blather.

pip install blather

Usage

from blather import Blather

blather = Blather()

# fine tunes an appropriate model on your dataset
blather.read("example_dataset.txt")

# returns a text sample generated from the model
blather.write('Sample text to complete')

# saves model
blather.save('model.pt')

# load model from previous training
blather.load('model.pt')

Contributing

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.

Please make sure to update tests as appropriate.

License

Apache