latest_news_backend_with_cat_pred_similar_news

Running

lalithadevi commited on Feb 14

Commit

88291b7

•

1 Parent(s): 8271332

Update news_extractor/news_extractor.py

Files changed (1) hide show

news_extractor/news_extractor.py CHANGED Viewed

@@ -143,5 +143,7 @@ def get_news():
     final_df.drop(columns=['elapsed_time'], inplace=True)
     final_df.drop_duplicates(subset='description', inplace=True)
     final_df = final_df.loc[(final_df["title"] != ""), :].copy()
     return final_df

     final_df.drop(columns=['elapsed_time'], inplace=True)
     final_df.drop_duplicates(subset='description', inplace=True)
     final_df = final_df.loc[(final_df["title"] != ""), :].copy()
+    final_df.loc[(final_df['description'].isna()) | (final_df['description']==''), 'description'] = final_df.loc[(final_df['description'].isna()) | (final_df['description']==''), 'title']
     return final_df