Spaces:
Sleeping
Sleeping
Update tasks/utils/predict.py
Browse files- 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 =
|
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
|