Update news_category_prediction.py

news_category_prediction.py

@@ -85,6 +85,10 @@ def predict_news_category(old_news: pd.DataFrame, new_news: pd.DataFrame, interp
             final_df.drop_duplicates(subset='url', keep='first', inplace=True)
         final_df.reset_index(drop=True, inplace=True)
         final_df.loc[final_df['pred_proba']<CLASSIFIER_THRESHOLD, 'category'] = 'OTHERS'
+        
+        if len(final_df) == 0:
+            final_df = None
+        
         logger.warning('Exiting predict_news_category()')
     except Exception as e:
         logger.warning(f'Unexcpected error in predict_news_category()\n{e}')