Text Generation AI - Next Word Prediction in Python