Update app.py

app.py

@@ -1,10 +1,17 @@
 import streamlit as st
+# from transformers import init_empty_weights
 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="Wi/arxiv-distilbert-base-cased")  # скачивание модели
+    return pipeline("sentiment-analysis")  # скачивание модели
 
 model = load_model()
 
@@ -84,6 +91,7 @@ if query:
             Most likely categories of the paper:
         </p>
     """, unsafe_allow_html=True)
-    result = format_predictions(top_pct(model(query)[0]))
-
+    # result = format_predictions(top_pct(model(query)[0]))
+    # result = query
+    result = model(query)[0]
     st.write(result)