LLM-leaderboard / main.py
geoalgo's picture
update
c708975
raw
history blame
981 Bytes
import random
from pathlib import Path
import gradio as gr
import pandas as pd
from gradio_leaderboard import Leaderboard, SelectColumns, SearchColumns
abs_path = Path(__file__).parent
df = pd.read_csv("opensci-ref-table.csv")
df.drop("#Tokens", axis=1, inplace=True)
df.drop("AVG", axis=1, inplace=True)
benchmarks = df.columns[1:]
df["Average ⬆️"] = df.loc[:, benchmarks].mean(axis=1)
# df.set_index("Model", inplace=True)
with gr.Blocks() as demo:
gr.Markdown(
"""
# πŸ₯‡ OpenEuroLLM Leaderboard πŸ‡ͺπŸ‡Ί
"""
)
Leaderboard(
value=df.round(2),
select_columns=SelectColumns(
default_selection=list(df.columns),
cant_deselect=["Model"],
label="Select Columns to Display:",
),
search_columns=SearchColumns(
primary_column="Model",
label="Filter a model",
secondary_columns=[],
),
)
if __name__ == "__main__":
demo.launch()