from simplemma import text_lemmatizer def lemmatizer_func(plain_text): words = text_lemmatizer(plain_text, lang="en") lemmatized_promt = ' '.join(words) return lemmatized_promt