Spaces:

mbabanov
/

devops

Runtime error

mbabanov commited on Apr 16, 2023

Commit

4829d4c

•

1 Parent(s): 7692f94

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import transformers
 st.markdown("### Articles classificator.")
 # st.markdown("<img width=200px src='https://rozetked.me/images/uploads/dwoilp3BVjlE.jpg'>", unsafe_allow_html=True)
-@st.cache_data
 def LoadModel():
     return torch.load('model.pt'), AutoTokenizer.from_pretrained('bert-base-uncased')()

 st.markdown("### Articles classificator.")
 # st.markdown("<img width=200px src='https://rozetked.me/images/uploads/dwoilp3BVjlE.jpg'>", unsafe_allow_html=True)
+@st.cache
 def LoadModel():
     return torch.load('model.pt'), AutoTokenizer.from_pretrained('bert-base-uncased')()