Update app.py

app.py

@@ -53,6 +53,7 @@ elif input_method == "Write or Paste New Text":
     st.subheader("Text to Run")
     input_text = st.text_area('Write or Paste Text Below', value="", height=128, max_chars=None, key=2)
 
+@st.cache(allow_output_mutation=True)
 def setModel(model_checkpoint, aggregation):
     model = AutoModelForTokenClassification.from_pretrained(model_checkpoint)
     tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)