Update app.py

app.py

@@ -12,17 +12,17 @@ st.markdown("Сервис классифицирует статьи по наз
 title = st.text_area("Введите название статьи")
 
 abstract = st.text_area("Введите аннотацию к статье, abstract статьи")
-
-model, tokenizer = load_model()
-#model.load_state_dict(torch.load('weight_model'))
-
-text = get_text(title, abstract)
-if text:
-  raw_predictions = get_labels(text, model, tokenizer)
-  st.markdown("Список классов к которым может относится данная статья")
-  for raw in raw_predictions:
-    st.markdown(f"{raw}")
-else:
-  st.markdown("Ваш запрос пуст. Введите хотя бы название")
+if st.button("Классифицируй!"):
+  model, tokenizer = load_model()
+  #model.load_state_dict(torch.load('weight_model'))
+  
+  text = get_text(title, abstract)
+  if text:
+    raw_predictions = get_labels(text, model, tokenizer)
+    st.markdown("Список классов к которым может относится данная статья")
+    for raw in raw_predictions:
+      st.markdown(f"{raw}")
+  else:
+    st.markdown("Ваш запрос пуст. Введите хотя бы название")