Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -45,16 +45,16 @@ default_text = '''질병관리청은 23일 지방자치단체가 보건당국과
|
|
45 |
topics_raw = ['IT/과학', '경제', '문화', '미용/건강', '사회', '생활', '스포츠', '연예', '정치']
|
46 |
|
47 |
|
48 |
-
|
49 |
-
|
50 |
|
51 |
name = st.sidebar.selectbox('Model', ['Topic Classification', 'Date Prediction'])
|
52 |
if name == 'Topic Classification':
|
53 |
title = 'News Topic Classification'
|
54 |
-
model, tokenizer =
|
55 |
elif name == 'Date Prediction':
|
56 |
title = 'News Date prediction'
|
57 |
-
model, tokenizer =
|
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 |
|