Update app.py

app.py

@@ -6,7 +6,7 @@ from torch import nn
 st.markdown("### Articles classificator.")
 # st.markdown("<img width=200px src='https://rozetked.me/images/uploads/dwoilp3BVjlE.jpg'>", unsafe_allow_html=True)
 
-@st.cache
+@st.cache(allow_output_mutation=True)
 def get_bert_and_tokenizer():
     model_name = 'bert-base-uncased'
     # return AutoModel.from_pretrained(model_name), AutoTokenizer.from_pretrained(model_name)