Fixed empty token issues in train_gensim.py