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
|