Update app.py

app.py

@@ -51,13 +51,16 @@ class RobertaClass(torch.nn.Module):
         output = self.classifier(pooler)
         return output
 
-tokenizer = RobertaTokenizer.from_pretrained('roberta-base', truncation=True, do_lower_case=True,
+@st.cache_resource
+def load_model():
+    tokenizer = RobertaTokenizer.from_pretrained('roberta-base', truncation=True, do_lower_case=True,
                                                  vocab_file='model/vocab.json',
                                                  merges_file='model/merges.txt')
+    model = torch.load('model/pytorch_roberta_sentiment.bin', map_location=torch.device('cpu'))
+    return model, tokenizer
 
 
-
-model = torch.load('model/pytorch_roberta_sentiment.bin', map_location=torch.device('cpu'))
+model, tokenizer = load_model()
 
 
 st.markdown("### Угадыватель")