in train.py aligning the text to lowercase