IlyaUsmanov commited on
Commit
d2be1e7
1 Parent(s): c9e28be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,13 +9,13 @@ def create_model():
9
  model.load_state_dict(m_state_dict)
10
  return model
11
 
12
- st.markdown("### arXiv paper classification!")
13
  tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-cased')
14
 
15
  model = create_model()
16
 
17
- title = st.text_area("Paper title")
18
- summary = st.text_area("Paper abstract(optional)", help="a.k.a. summary")
19
  if st.button("Submit"):
20
  if title == "":
21
  st.error("Please enter your paper title")
 
9
  model.load_state_dict(m_state_dict)
10
  return model
11
 
12
+ st.markdown("### arXiv paper tag classification!")
13
  tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-cased')
14
 
15
  model = create_model()
16
 
17
+ title = st.text_area("Enter your paper title")
18
+ summary = st.text_area("Enter your paper abstract(optional)", help="a.k.a. summary")
19
  if st.button("Submit"):
20
  if title == "":
21
  st.error("Please enter your paper title")