Update app.py

app.py

@@ -190,7 +190,7 @@ def classify_text(title, description, show_all=False, threshold=0.95):
         filtered_results = []
         for i in results[0]:
             filtered_results.append((i['label'], i['score']))
-            cumulative_prob += score
+            cumulative_prob += i['score']
             if cumulative_prob >= threshold:
                 break
         return filtered_results
@@ -205,7 +205,7 @@ description = st.text_area("Краткое описание статьи", heigh
 
 # Кнопка "Классифицировать"
 if st.button("Классифицировать"):
-    if not title or not description:
+    if not title and not description:
         st.warning("Пожалуйста, заполните хотя бы одно поле.")
     else:
         with st.spinner("Идет классификация..."):  # Индикатор загрузки