Update news_category_similar_news_prediction.py

news_category_similar_news_prediction.py

@@ -121,6 +121,7 @@ def add_flags_to_prediction_df(df):
     df.drop(columns=['similar_news', 'src', 'title', 'description'], inplace=True)
     df['y_true_proba'] = df['y_pred_proba']
     df['y_true'] = df['y_pred']
+    df['updated_prediction'] = 1
     return df