Spaces:
Running
Running
Update app.py
Browse files
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.
|
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(
|