Update news_category_similar_news_prediction.py

news_category_similar_news_prediction.py

@@ -157,7 +157,7 @@ def predict_news_category_similar_news(old_news: pd.DataFrame, new_news: pd.Data
                 
                 headlines = [*new_news['title']].copy()
 
-                headlines_desc = [*final_df['title'] + ". " + final_df['description']].copy()
+                headlines_desc = [*new_news['title'] + ". " + new_news['description']].copy()
                 label, prob = inference(headlines_desc, interpreter, label_encoder, tokenizer)