Train and use generative text models in a few lines of code.
To see blather in action check out the colab notebook!
Use the package manager pip to install blather.
pip install blatherfrom 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')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.