Update app.py

app.py

@@ -7,6 +7,7 @@ from transformers import TrainingArguments, \
 categories = ['Biology', 'Computer science', 'Economics', 'Electrics', 'Finance',
                'Math', 'Physics', 'Statistics']
 labels = [i for i in range(len(categories))]
+
 def print_probs(logits):
   probs = torch.nn.functional.softmax(logits, dim=0).numpy()*100
   ans = list(zip(probs,labels))
@@ -19,7 +20,8 @@ def print_probs(logits):
     st.write(text)
     sum+=prob
     i+=1
-
+    
+@st.cache
 def make_prediction(text):
   tokenized_text = tokenizer(text, return_tensors='pt')
   with torch.no_grad():
@@ -54,6 +56,6 @@ text = st.text_area("Введите название статьи", height=50)
 st.markdown("### Article Abstract")
 text = st.text_area("Введите описание статьи", height=400)
 
-@st.cache
+
 make_prediction(text)