Update app.py

app.py

@@ -62,6 +62,7 @@ elif context == "New Context":
     selected_context = context_3
     selected_question = st.text_area("New Question", value="", height=64)
 
+@st.cache(allow_output_mutation=True)
 def setModel(model_checkpoint):
     model = AutoModelForQuestionAnswering.from_pretrained(model_checkpoint)
     tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)