kxx-kkk commited on
Commit
0f41a29
1 Parent(s): 787cf57

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ st.subheader("Try")
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, task="question-answering")
 
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_data(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, task="question-answering")