Update news_category_similar_news_prediction.py

news_category_similar_news_prediction.py

@@ -203,6 +203,10 @@ def predict_news_category_similar_news(old_news: pd.DataFrame, new_news: pd.Data
 
                 final_df = process_prediction_df(final_df, df_type="production")
                 prediction_df = new_news.copy()
+                prediction_df.loc[(prediction_df['title'].str.contains('Pakistan')) & (prediction_df['category'] == 'NATION'), 'category'] = 'WORLD'
+                logger.warning('Updated category of articles having Pakistan in title and category=NATION to WORLD in prediction_df')
+                prediction_df.loc[(prediction_df['title'].str.contains('Zodiac Sign', case=False)) | (prediction_df['title'].str.contains('Horoscope', case=False)), 'category'] = 'ASTROLOGY'
+                logger.warning('Updated category of articles having Zodiac Sign in title to ASTROLOGY in prediction_df')
                 prediction_df = add_flags_to_prediction_df(prediction_df)
                 
             else: