Thank you Philipp. I will make this changes. It will be great if you guys can create a notebook example / post for this type of task.
In terms of data set up, I reviewed the training script and correct me if I am wrong but I did not see any requirements to add start text or end text prefixes to the data? Just a column name text with the input text data?
Thank you again. Great job as always