Update tasks/text.py

tasks/text.py

@@ -12,6 +12,8 @@ import joblib
 REPO_ID = "kantundpeterpan/frugal-ai-toy"
 FILENAME = "tfidf_rf.pkl"
 
+def lemmatize_X(X):
+    return X.quote.apply(tokenize_quote).apply(lemmatize_tokens).apply(lambda x: " ".join(x))
 
 import random