Update tasks/utils/preprocessing.py

tasks/utils/preprocessing.py

@@ -4,6 +4,7 @@ import string
 from nltk.corpus import stopwords
 import nltk
 import spacy
+import pandas as pd
 
 nltk.download('stopwords')
 # Get the list of English stop words from NLTK
@@ -36,7 +37,7 @@ def process_text(text):
     return processed_tokens
 
     
-def predict(input_df: pd.DataFrame, tfidf_path: str, model_path: str, text_column: str = "quote"):
+def predict(input_df: pd.DataFrame, tfidf_path: str, model_path: str):
     """
     Predict the output using a saved TF-IDF vectorizer and Random Forest model.