Update app.py

app.py

@@ -154,7 +154,7 @@ except OSError as e:
     st.stop()  # Остановка выполнения приложения при ошибке
 
 
-def classify_text(title, description, show_all=False, threshold=0.95):
+def classify_text(title, description):
     """
     Классифицирует текст и возвращает результаты в отсортированном виде.
 
@@ -170,7 +170,6 @@ def classify_text(title, description, show_all=False, threshold=0.95):
     text = f"{title} {description}"  # Объединяем заголовок и описание
     topic_classifier = pipeline("text-classification", model=model, tokenizer=tokenizer, top_k = len(id_to_cat))
     try:
-        
         results = topic_classifier(text) 
         # results = topic_classifier(text, candidate_labels, multi_label=True)  # multi_label=True для нескольких меток
     except Exception as e:
@@ -180,24 +179,16 @@ def classify_text(title, description, show_all=False, threshold=0.95):
     for i in results[0]:
         i['label'] = id_to_cat[int(i['label'].split('_')[1])]
 
-    if show_all:
-        filtered_results = []
-        for i in results[0]:
-            filtered_results.append((i['label'], i['score']))
-        return filtered_results
-    else:
-        cumulative_prob = 0
-        filtered_results = []
-        for i in results[0]:
-            filtered_results.append((i['label'], i['score']))
-            cumulative_prob += i['score']
-            if cumulative_prob >= threshold:
-                break
-        return filtered_results
+
+    filtered_results = []
+    for i in results[0]:
+        filtered_results.append((i['label'], i['score']))
+    return filtered_results
+        
 
 
 # --- Интерфейс Streamlit ---
-st.title("Классификация статей 11")
+st.title("Классификация статей 1")
 
 # Ввод данных
 title = st.text_input("Заголовок статьи")
@@ -211,18 +202,21 @@ if st.button("Классифицировать"):
         with st.spinner("Идет классификация..."):  # Индикатор загрузки
             results = classify_text(title, description)
             if results:
-              st.subheader("Результаты классификации (с ограничением по вероятности):")
+              st.subheader("Результаты классификации (top 95%):")
+              cumulative_prob = 0
               for label, score in results:
                   st.write(f"- **{label}**: {score:.4f}")
+                  cumulative_prob += score
+                  if cumulative_prob >= 0.95:
+                      break
 
               # Кнопка "Показать все"
               if st.button("Показать все категории"):
-                  all_results = classify_text(title, description, candidate_labels, show_all=True)
                   st.subheader("Полные результаты классификации:")
-                  for label, score in all_results:
+                  for label, score in results:
                       st.write(f"- **{label}**: {score:.4f}")
             else:
                 st.info("Не удалось получить результаты классификации.")
 
 elif title or description: #небольшой костыль, чтобы при старте не было предупреждения
-    st.warning("Пожалуйста, заполните все поля.")
+    st.warning("Пожалуйста, заполните хотя бы одно поле.")