Update app.py

app.py

@@ -3,6 +3,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from huggingface_hub import hf_hub_download
 import torch
 import numpy as np
+import pandas as pd
 import json
 
 @st.cache_resource
@@ -52,6 +53,15 @@ if st.button("Classify"):
 
         top_labels = get_top95(id2label, probs)
 
-        st.markdown("Top 95% topics:")
+        results = []
         for label, prob in top_labels:
-            st.markdown(f"- **{categories[label]} ({label})**: {prob * 100:.1f}%")
+            results.append({
+                "Category": categories[label],
+                "ID": label,
+                "Confidence": f"{prob * 100:.1f} %"
+            })
+        
+        df = pd.DataFrame(results)
+        
+        st.markdown("### 🧠 Top 95% Predicted Topics")
+        st.dataframe(df, use_container_width=True)