Two files to interact with are "SRNTrain.py" and "loadingModel.py". The former reads in a body of text which the model is trained on (modify the param to change between source options), and the latter calls upon that model to predict a new body of text based off a prompt string.
These models were trained on
- Karl Marx's Communist Manifesto, sourced from: https://www.gutenberg.org/files/61/61.txt
- Spam email I receive 📬