OOlajide commited on
Commit
7be3111
1 Parent(s): 74880f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,11 +12,11 @@ expander = st.sidebar.expander("About")
12
  expander.write("This web app allows you to perform common Natural Language Processing tasks, select a task below to get started.")
13
 
14
  st.sidebar.header("What will you like to do?")
15
- option = st.sidebar.radio("", ["Extractive question answering", "Text summarization", "Text generation"])
16
 
17
  @st.cache(show_spinner=False, allow_output_mutation=True)
18
  def question_model():
19
- model_name = "deepset/roberta-base-squad2"
20
  question_answerer = pipeline(model=model_name, tokenizer=model_name, revision="v1.0", task="question-answering")
21
  return question_answerer
22
 
 
12
  expander.write("This web app allows you to perform common Natural Language Processing tasks, select a task below to get started.")
13
 
14
  st.sidebar.header("What will you like to do?")
15
+ option = st.sidebar.radio("", ["Text summarization", "Extractive question answering", "Text generation"])
16
 
17
  @st.cache(show_spinner=False, allow_output_mutation=True)
18
  def question_model():
19
+ model_name = "deepset/tinyroberta-squad2"
20
  question_answerer = pipeline(model=model_name, tokenizer=model_name, revision="v1.0", task="question-answering")
21
  return question_answerer
22