alex6095 commited on
Commit
20f3361
1 Parent(s): 27801b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -45,16 +45,16 @@ default_text = '''질병관리청은 23일 지방자치단체가 보건당국과
45
  topics_raw = ['IT/과학', '경제', '문화', '미용/건강', '사회', '생활', '스포츠', '연예', '정치']
46
 
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 = get_topic()
55
  elif name == 'Date Prediction':
56
  title = 'News Date prediction'
57
- model, tokenizer = get_date()
58
 
59
  st.title(title)
60
 
 
45
  topics_raw = ['IT/과학', '경제', '문화', '미용/건강', '사회', '생활', '스포츠', '연예', '정치']
46
 
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
55
  elif name == 'Date Prediction':
56
  title = 'News Date prediction'
57
+ model, tokenizer = date_model, date_tokenizer
58
 
59
  st.title(title)
60