Update app.py

app.py

@@ -1,10 +1,12 @@
 import streamlit as st
-from transformers import pipeline
-
+from transformers import pipeline, AutoModelForQuestionAnswering, AutoTokenizer
 @st.cache(allow_output_mutation=True)
 def load_qa_model():
-    model = pipeline("question-answering")
-    return model
+    model_name = "mrm8488/mobilebert-uncased-finetuned-squadv2"
+    model = AutoModelForQuestionAnswering.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    qa = pipeline("question-answering", model=model, tokenizer=tokenizer)
+    return qa
 
 qa = load_qa_model()
 st.title("Ask Questions about your Text")