albertvillanova's picture
Update app.py
a991678 verified
raw
history blame contribute delete
748 Bytes
import gradio as gr
import pandas as pd
import plotly.express as px
DATA = {"BBH": [0.2, 0.7], "GPQA": [0.4, 0.5], "IFEval": [0.6, 0.3], "MATH": [0.2, 0.7], "MMLU-Pro": [0.4, 0.5], "MuSR": [0.6, 0.3]}
def display_plot():
df = pd.DataFrame(DATA, index=["model-1", "model-2"])
fig = px.line_polar(
df.melt(ignore_index=False, var_name="Benchmark", value_name="Score").reset_index(names="Model"),
r="Score", theta="Benchmark", color="Model",
line_close=True,
range_r=[0, 1],
color_discrete_sequence=["#FF9D00", "#32343D"],
)
return fig
with gr.Blocks() as demo:
plot = gr.Plot()
btn = gr.Button()
btn.click(
fn=display_plot,
outputs=plot,
)
demo.launch()