lalithadevi commited on
Commit
1bbda15
1 Parent(s): b9da9ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -42,7 +42,11 @@ def update_news():
42
  db_read = DBRead()
43
  db_write = DBWrite()
44
  old_news = db_read.read_news_from_db()
 
 
45
  new_news = get_news()
 
 
46
  news_df = predict_news_category(old_news, new_news, interpreter, label_encoder, tokenizer)
47
  news_json = [*json.loads(news_df.reset_index(drop=True).to_json(orient="index")).values()]
48
  db_write.insert_news_into_db(news_json)
 
42
  db_read = DBRead()
43
  db_write = DBWrite()
44
  old_news = db_read.read_news_from_db()
45
+ if old_news is None:
46
+ logger.warning('No old news Found')
47
  new_news = get_news()
48
+ if new_news is None:
49
+ raise Exception('No news news Found')
50
  news_df = predict_news_category(old_news, new_news, interpreter, label_encoder, tokenizer)
51
  news_json = [*json.loads(news_df.reset_index(drop=True).to_json(orient="index")).values()]
52
  db_write.insert_news_into_db(news_json)