sumesh4C commited on
Commit
c9878b6
·
verified ·
1 Parent(s): 944b37c

Update tasks/utils/predict.py

Browse files
Files changed (1) hide show
  1. tasks/utils/predict.py +1 -1
tasks/utils/predict.py CHANGED
@@ -31,7 +31,7 @@ def predict(input_df: pd.DataFrame, tfidf_path:str , model_path: str):
31
  with open(model_path, "rb") as model_file:
32
  model = pickle.load(model_file)
33
 
34
- tfidf_vectorizer = = TfidfVectorizer(**params)
35
  tfidf_vectorizer.set_params(preprocessor=process_text)
36
 
37
  # Transform the input text using the TF-IDF vectorizer
 
31
  with open(model_path, "rb") as model_file:
32
  model = pickle.load(model_file)
33
 
34
+ tfidf_vectorizer = TfidfVectorizer(**params)
35
  tfidf_vectorizer.set_params(preprocessor=process_text)
36
 
37
  # Transform the input text using the TF-IDF vectorizer