Update app.py

app.py

@@ -20,9 +20,9 @@ def load_tokenizer():
     return AutoTokenizer.from_pretrained('microsoft/deberta-v3-small')
 
 
-#model = load_model()
-#id2classes = load_id2classes()
-#tokenizer = load_tokenizer()
+model = load_model()
+id2classes = load_id2classes()
+tokenizer = load_tokenizer()
 
 
 def get_top_classes(text):
@@ -48,10 +48,9 @@ if st.button('Submit'):
         else:
             text = title.title()
 
-        #top_classes, probs = get_top_classes(text)
+        top_classes, probs = get_top_classes(text)
 
-        #for p, cls in zip(probs, top_classes):
-            #st.success(f'Статья относится к {cls} с вероятностью {p}')
-        st.success('Запустилось')
+        for p, cls in zip(probs, top_classes):
+            st.success(f'Статья относится к {cls} с вероятностью {p}')
     else:
         st.error('Введите название статьи')