Update app.py

app.py

@@ -1,7 +1,7 @@
 from news_extractor import get_news
 from db_operations.db_write import DBWrite
 from db_operations.db_read import DBRead
-from news_category_prediction import predict_news_category
+from news_category_similar_news_prediction import predict_news_category_similar_news
 import json
 from flask import Flask, Response
 from flask_cors import cross_origin, CORS
@@ -48,7 +48,8 @@ def update_news():
         db_write = DBWrite()
         old_news = db_read.read_news_from_db()
         new_news = get_news()
-        news_df, is_db_updation_required = predict_news_category(old_news, new_news, interpreter, label_encoder, tokenizer)
+        news_df, is_db_updation_required = predict_news_category_similar_news(old_news, new_news, interpreter, label_encoder, 
+                                                                              tokenizer, collection, vectorizer, sent_model, ce_model)
         if news_df is None:
             raise Exception('Could not generate category predictions. Aborting the database insertion. No new articles are inserted into the collection.')