Add download button for table + update todos

README.md

@@ -78,15 +78,12 @@ We are always happy for contributions! You can contribute by the following:
 
 ## Future Ideas
 
-- add model year
-- add "export current view as .csv" button to streamlit demo
+- (TBD) add model year
 - (TBD) add model details:
     - #params
     - #tokens seen during training
     - length context window
     - architecture type (transformer-decoder, transformer-encoder, transformer-encoder-decoder, ...)
-- if additional model details, allow to hide them in the interactive streamlit dashboard with a checkbox?
-- (TBD) improvements on the filtering in the streamlit demo, maybe filter by value range?
 
 ## More Open LLMs
 

streamlit_app.py

@@ -4,8 +4,7 @@
 
 import pandas as pd
 import streamlit as st
-from pandas.api.types import (is_bool_dtype, is_datetime64_any_dtype,
-                              is_numeric_dtype)
+from pandas.api.types import is_bool_dtype, is_datetime64_any_dtype, is_numeric_dtype
 
 GITHUB_URL = "https://github.com/LudwigStumpp/llm-leaderboard"
 
@@ -219,6 +218,22 @@ def setup_leaderboard(readme: str):
 
     st.dataframe(df_leaderboard)
 
+    st.download_button(
+        "Download current selection as .html",
+        df_leaderboard.to_html().encode("utf-8"),
+        "leaderboard.html",
+        "text/html",
+        key="download-html",
+    )
+
+    st.download_button(
+        "Download current selection as .csv",
+        df_leaderboard.to_csv().encode("utf-8"),
+        "leaderboard.csv",
+        "text/csv",
+        key="download-csv",
+    )
+
 
 def setup_benchmarks(readme: str):
     benchmarks_table = extract_markdown_table_from_multiline(readme, table_headline="## Benchmarks")