Update app.py

app.py

@@ -15,11 +15,16 @@ title = st.text_area("TITLE HERE")
 abstract = st.text_area("ABSTRACT HERE")
 # ^-- показать текстовое поле. В поле text лежит строка, которая находится там в данный момент
 
-tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
-model = BertForSequenceClassification.from_pretrained(
-    "bert-base-uncased", # Use the 12-layer BERT model, with an uncased vocab.
-    num_labels = 44,)
-model.load_state_dict(torch.load("model_last_version.pt", map_location=torch.device('cpu')))
+@st.cache_data
+def load_model_and_tokenizer():    
+    tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
+    model = BertForSequenceClassification.from_pretrained(
+        "bert-base-uncased", # Use the 12-layer BERT model, with an uncased vocab.
+        num_labels = 44,)
+    model.load_state_dict(torch.load("model_last_version.pt", map_location=torch.device('cpu')))
+    return model, tokenizer
+
+model, tokenizer = load_model_and_tokenizer()
 MAX_LEN = 64
 # Преобразуем название статьи в токены
 tokens = tokenizer(title, padding=True, truncation=True, return_tensors="pt")