Update news_category_similar_news_prediction.py

news_category_similar_news_prediction.py

@@ -75,7 +75,7 @@ def predict_news_category_similar_news(old_news: pd.DataFrame, new_news: pd.Data
             final_df.drop_duplicates(subset='url', keep='first', inplace=True)
             headlines = [*final_df['title']].copy()
             label, prob = inference(headlines, interpreter, label_encoder, tokenizer)
-            sent_embs = vectorizer.vectorize_(headlines)
+            sent_embs = vectorizer.vectorize_(headlines, sent_model)
             sim_news = [find_similar_news(search_vec, collection, vectorizer, sent_model, ce_model) for search_vec in sent_embs]
             final_df['category'] = label
             final_df['pred_proba'] = prob
@@ -91,7 +91,7 @@ def predict_news_category_similar_news(old_news: pd.DataFrame, new_news: pd.Data
             if len(new_news) > 0:
                 headlines = [*new_news['title']].copy()
                 label, prob = inference(headlines, interpreter, label_encoder, tokenizer)
-                sent_embs = vectorizer.vectorize_(headlines)
+                sent_embs = vectorizer.vectorize_(headlines, sent_model)
                 sim_news = [find_similar_news(search_vec, collection, vectorizer, sent_model, ce_model) for search_vec in sent_embs]
                 new_news['category'] = label
                 new_news['pred_proba'] = prob