SatAT commited on
Commit
b050d1c
·
1 Parent(s): ed2bd2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from torch.utils.data import TensorDataset, DataLoader, RandomSampler, Sequentia
10
  st.markdown("### Paper category classification")
11
  st.markdown("<img width=200px src='https://grandgames.net/img/upload/0d153888a24eb5b8c0195495cd83d0dd.jpg'>", unsafe_allow_html=True)
12
  # ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter
13
- @st.cache
14
  def load_model_and_tokenizer():
15
  tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
16
  model = BertForSequenceClassification.from_pretrained(
 
10
  st.markdown("### Paper category classification")
11
  st.markdown("<img width=200px src='https://grandgames.net/img/upload/0d153888a24eb5b8c0195495cd83d0dd.jpg'>", unsafe_allow_html=True)
12
  # ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter
13
+ @st.cache_data
14
  def load_model_and_tokenizer():
15
  tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
16
  model = BertForSequenceClassification.from_pretrained(