Update app.py

app.py

@@ -5,7 +5,7 @@ import re
 
 # Constants
 GITHUB_URL = "https://github.com/Sartify/STEL"
-NON_BENCHMARK_COLS = ["Open?", "Publisher", "Basemodel", "Matryoshka", "Dimension"]
+POSSIBLE_NON_BENCHMARK_COLS = ["Open?", "Publisher", "Basemodel", "Matryoshka", "Dimension", "Model Name"]
 
 def extract_table_from_markdown(markdown_text, table_start):
     """Extract table content from markdown text."""
@@ -34,18 +34,21 @@ def setup_page():
     """Set up the Streamlit page."""
     st.set_page_config(page_title="Swahili Text Embeddings Leaderboard", page_icon="⚡", layout="wide")
     st.title("⚡ Swahili Text Embeddings Leaderboard (STEL)")
-    st.image("STEL.jpg", width=300)
+    st.image("https://raw.githubusercontent.com/username/repo/main/STEL.jpg", width=300)
 
 def display_leaderboard(df):
     """Display the leaderboard."""
     st.header("📊 Leaderboard")
     
+    # Determine which non-benchmark columns are present
+    present_non_benchmark_cols = [col for col in POSSIBLE_NON_BENCHMARK_COLS if col in df.columns]
+    
     # Add filters
-    columns_to_filter = [col for col in df.columns if col not in NON_BENCHMARK_COLS]
+    columns_to_filter = [col for col in df.columns if col not in present_non_benchmark_cols]
     selected_columns = st.multiselect("Select benchmarks to display:", columns_to_filter, default=columns_to_filter)
     
     # Filter dataframe
-    df_display = df[NON_BENCHMARK_COLS + selected_columns]
+    df_display = df[present_non_benchmark_cols + selected_columns]
     
     # Display dataframe
     st.dataframe(df_display)
@@ -142,4 +145,6 @@ def main():
     st.markdown("Thank you for being part of this effort to advance Swahili language technologies!")
 
 if __name__ == "__main__":
-    main()
+    main()
+
+