ds_visualiser / app.py
amaljoe-ibm's picture
update filename
9523f33
raw
history blame contribute delete
794 Bytes
import gradio as gr
import pandas as pd
dataset_files = {
"XTREME Overview": "xtreme-overview-merged.csv",
}
def load_dataset(name):
df = pd.read_csv(dataset_files[name])
# df.reset_index(inplace=True)
return df
with gr.Blocks() as demo:
gr.Markdown("# TLP matrix: Languages x Tasks")
gr.Markdown("Counts of dataset sizes per language and task")
dataset_dropdown = gr.Dropdown(choices=list(dataset_files.keys()), label="Select dataset")
# Set initial dataframe value here
initial_dataset = list(dataset_files.keys())[0]
initial_df = load_dataset(initial_dataset)
dataframe = gr.Dataframe(value=initial_df, type="pandas", interactive=False)
dataset_dropdown.change(fn=load_dataset, inputs=dataset_dropdown, outputs=dataframe)
demo.launch()