Update tasks/utils/predict.py

tasks/utils/predict.py

@@ -8,7 +8,7 @@ from tasks.utils.preprocessing import process_text
 import json
 from sklearn.feature_extraction.text import TfidfVectorizer
     
-def predict(input_df: pd.DataFrame, tfidf_vectorizer:str , model_path: str):
+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.