Update CustomLanguageModel.py