Update tasks/utils/predict.py

tasks/utils/predict.py

@@ -6,7 +6,7 @@ import sys
 sys.path.append(".") 
 from tasks.utils.preprocessing import process_text
     
-def predict(input_df: pd.DataFrame, tfidf_path: str, model_path: str):
+def predict(input_df: pd.DataFrame, tfidf_vectorizer , model_path: str):
     """
     Predict the output using a saved TF-IDF vectorizer and Random Forest model.
 
@@ -19,9 +19,11 @@ def predict(input_df: pd.DataFrame, tfidf_path: str, model_path: str):
     Returns:
         pd.Series: Predictions for each row in the input dataframe.
     """
+    """
     # Load the TF-IDF vectorizer
     with open(tfidf_path, "rb") as tfidf_file:
         tfidf_vectorizer = pickle.load(tfidf_file)
+    """
 
     # Load the Random Forest model
     with open(model_path, "rb") as model_file: