text_plain_output_1.png
from aitextgen import aitextgen

ai2 = aitextgen(model_folder='./trained_model', tokenizer_file='aitextgen.tokenizer.json')
ai2.generate(10, prompt='<START>\nTHERAPIST:\nHi, Emily.\nCLIENT:\nHi. I am feeling low and have been drinking alcohol every day. What do you think that is happening\n')