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