Commit
•
1bbda15
1
Parent(s):
b9da9ca
Update app.py
Browse files
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)
|