gitmylo / TextGenerationMLpy

Text generation ML

a machine learning model that generates text based on a given input.

installation/setup

make sure to run pip install -r requirements.txt and that you have a cuda compatible gpu.

usage

• create a folder named data and put your text files in there
• simply run python main.py for an example with training and generation.
• use model.py as a library for training and generation in your own projects.