Update app.py

app.py

@@ -3,17 +3,11 @@ import streamlit as st
 from transformers import pipeline
 
 
-# @st.cache_resource  # кэширование
-# def load_model():
-#     return pipeline("text-classification", model="Wi/arxiv-distilbert-base-cased")  # скачивание модели
-#
-# model = load_model()
+@st.cache_resource  # кэширование
+def load_model():
+    return pipeline("text-classification", model="voroninip/bert-paper-classifier-arxiv", top_k=None)  # скачивание модели
 
-# @st.cache_resource  # кэширование
-# def load_model():
-#     return pipeline("sentiment-analysis")  # скачивание модели
-
-# model = load_model()
+model = load_model()
 
 def top_pct(preds, threshold=.95):
     preds = sorted(preds, key=lambda x: -x["score"])
@@ -91,7 +85,5 @@ if query:
             Most likely categories of the paper:
         </p>
     """, unsafe_allow_html=True)
-    # result = format_predictions(top_pct(model(query)[0]))
-    # result = query
-    # result = model(query)[0]
+    result = format_predictions(top_pct(model(query)[0]))
     st.write(result)