Update app.py

app.py

@@ -31,6 +31,7 @@ def make_prediction(text):
 
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 
+@st.cache
 model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
 model_name = "trained_model2"
 model_path = model_name + '.zip'
@@ -49,12 +50,11 @@ st.markdown("<img src='https://centroderecursosmarista.org/wp-content/uploads/20
 st.markdown(" ")
 
 st.markdown("### Article Title")
-text = st.text_area("Введите название статьи", height=400)
+text = st.text_area("Введите название статьи", height=50)
 
 st.markdown("### Article Abstract")
 text = st.text_area("Введите описание статьи", height=400)
 
-
-
+@st.cache
 make_prediction(text)