alex6095 commited on
Commit
4e2e011
1 Parent(s): 20f3361

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -47,8 +47,10 @@ topics_raw = ['IT/과학', '경제', '문화', '미용/건강', '사회', '생
47
 
48
  topic_model, topic_tokenizer = get_topic()
49
  date_model, date_tokenizer = get_date()
 
50
 
51
  name = st.sidebar.selectbox('Model', ['Topic Classification', 'Date Prediction'])
 
52
  if name == 'Topic Classification':
53
  title = 'News Topic Classification'
54
  model, tokenizer = topic_model, topic_tokenizer
 
47
 
48
  topic_model, topic_tokenizer = get_topic()
49
  date_model, date_tokenizer = get_date()
50
+ st.sidebar.header('Menu')
51
 
52
  name = st.sidebar.selectbox('Model', ['Topic Classification', 'Date Prediction'])
53
+
54
  if name == 'Topic Classification':
55
  title = 'News Topic Classification'
56
  model, tokenizer = topic_model, topic_tokenizer