Fix weights table rendering

app.py

@@ -298,17 +298,13 @@ if st.button("🔍 Detect", type="primary"):
 
             # Ensemble weights
             st.markdown("### Ensemble Weights")
-            weight_col1, weight_col2 = st.columns(2)
-            with weight_col1:
-                st.markdown("| Model | Weight | F1 |")
-                st.markdown("|---|---|---|")
-                st.markdown("| Logistic Regression | 17.0% | 0.818 |")
-                st.markdown("| SVM | 21.9% | 0.871 |")
-            with weight_col2:
-                st.markdown("| Model | Weight | F1 |")
-                st.markdown("|---|---|---|")
-                st.markdown("| Random Forest | 23.4% | 0.886 |")
-                st.markdown("| CodeBERT | 37.7% | 0.998 |")
+            weights_data = {
+                "Model": ["Logistic Regression", "SVM", "Random Forest", "CodeBERT"],
+                "Weight": ["17.0%", "21.9%", "23.4%", "37.7%"],
+                "F1 Score": ["0.818", "0.871", "0.886", "0.998"],
+            }
+            import pandas as pd
+            st.table(pd.DataFrame(weights_data))
 
             # Features
             st.markdown("### Key Features Extracted")

requirements.txt

@@ -2,4 +2,5 @@ torch==2.5.1
 transformers==4.41.2
 streamlit==1.32.0
 numpy==1.26.4
-scikit-learn==1.3.2
+scikit-learn==1.3.2
+pandas==2.1.4