Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer, AutoModelForQuestionAnswering
 
 st.set_page_config(page_title="Common NLP Tasks")
 st.title("Common NLP Tasks")
@@ -13,7 +13,8 @@ option = st.sidebar.radio('', ['Extractive question answering', 'Text summarizat
 
 @st.cache(show_spinner=False, allow_output_mutation=True)
 def question_model():
-    question_answerer = pipeline("question-answering")
+    tokenizer = AutoTokenizer.from_pretrained("deepset/roberta-base-squad2")
+    model = AutoModelForQuestionAnswering.from_pretrained("deepset/roberta-base-squad2")
     return question_answerer
 
 @st.cache(show_spinner=False, allow_output_mutation=True)