Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import torch
 from transformers import DistilBertForSequenceClassification, DistilBertTokenizerFast
-from utils import get_text, get_label
+from utils import get_text, get_labels
 
 count_labels = 8
 
@@ -20,7 +20,9 @@ model.load_state_dict(torch.load('weight_model'))
 text = get_text(title, abstract)
 if text:
   raw_predictions = get_labels(text, model, tokenizer)
-  st.markdown(f"{raw_predictions}")
+  st.markdown("Список классов к которым может относится данная статья")
+  for raw in raw_predictions:
+    st.markdown(f"{raw}")
 else:
   st.markdown("Ваш запрос пуст. Введите хотя бы название")