File size: 1,280 Bytes
76050e6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
import streamlit as st
from draw_utils import PAGE_MARKDOWN, PAGE_INFO, LENGTHS
from draw_utils import load_results, style_dataframe
st.set_page_config(layout="wide", page_title="Leaderboard App")
st.markdown(PAGE_MARKDOWN, unsafe_allow_html=True)
def draw_leaderboard():
df = load_results()
tasks = ['avg'] + [f"qa{i}" for i in range(1, 11)]
columns = ["model_name", "avg(32k)", "avg(128k)"] + LENGTHS
st.title("🔎📚🪡📚❓ BABILong Leaderboard 🏆")
st.markdown(PAGE_INFO)
st.subheader("Average Accuracy")
search_term = st.text_input("Search models:", "")
tabs = st.tabs([str(task) for task in tasks])
for i, tab in enumerate(tabs):
with tab:
task_df = df[df.task == tasks[i]][columns]
if search_term:
task_df = task_df[task_df['model_name'].str.contains(search_term, case=False)]
task_df.reset_index(drop=True, inplace=True)
row_height = 35
height = (len(task_df) + 1) * row_height
styled = style_dataframe(task_df).format(precision=2)
st.dataframe(
styled,
use_container_width=True,
height=height,
)
if __name__ == "__main__":
draw_leaderboard()
|