In Text-Style-Transfer Example, how do you use the model on a chosen sentence?

Hi,
I tried running the code for the text style transfer example after reading the related paper(Unsupervised Text Style Transfer using Language Models as Discriminators) and I have a doubt regarding its use.
I am running the code main.py and it is training on the sample yelp datasets given. Training the model on my custom dataset seems simple enough, I just change the path in the config.py file, but after training, if I want to manually use the model to perform a text style transfer on certain sentences I've chosen and read the outputs, how would I go about doing that? Is there a function I can run with a string as an input that gives the transformed string as the output?
I apologize if this is too trivial, I'm very much beginner in this subject.
Thanks in advance for your help

Start by reading the text style transfer example in main.py and its configuration in config.py, then trace how training produces and consumes the model. Document the supported way to provide a chosen sentence and read the transformed output, or identify the missing entry point; done means a newcomer can follow the example for a single sentence after training.

Written by the indexing model from the issue text.