Update app.py

app.py

@@ -25,14 +25,14 @@ def get_top95(labels, probs):
     sorted_labels = [labels[i.item()] for i in sorted_indices]
 
     cumulative = torch.cumsum(sorted_probs, dim=0)
-    cutoff = torch.where(cumulative >= 0.8)[0]
+    cutoff = torch.where(cumulative >= 0.95)[0]
     last_idx = cutoff[0].item() + 1 if len(cutoff) > 0 else len(sorted_probs)
 
     return list(zip(sorted_labels[:last_idx], sorted_probs[:last_idx].tolist()))
 
 # UI
 st.set_page_config(page_title="Article Topic Classifier")
-st.title("🔬 Article Topic Classifier")
+st.title("Article Topic Classifier")
 st.markdown("Enter the **title** and optionally **abstract** of the article.")
 
 title = st.text_input("Title", placeholder="e.g. Neural Networks for Quantum Physics")
@@ -52,5 +52,6 @@ if st.button("Classify"):
 
         top_labels = get_top95(id2label, probs)
 
+        st.markdown("Top 95% topics:")
         for label, prob in top_labels:
             st.markdown(f"- **{categories[label]} ({label})**: {prob * 100:.1f}%")