lemma.py

import nltk
from nltk.stem import WordNetLemmatizer

# Initialize the lemmatizer
lemmatizer = WordNetLemmatizer()

def lemmatize_text(text):
    # Tokenize the text
    tokens = nltk.word_tokenize(text)

    # Lemmatize the tokens
    lemmas = [lemmatizer.lemmatize(token) for token in tokens]

    # Join the lemmas into a single string
    lemmatized_text = ' '.join(lemmas)

    return lemmatized_text
    
# Test the function
text = "I am running to the store to buy milk"
lemmatize_text(text)