import gradio as gr
import pandas as pd

# df = pd.read_table("visit_bench_leaderboard.tsv")
df = pd.read_table('visitbench_leaderboard_Single~Image_Oct282023.tsv')

headline = """# VisIT-Bench Leaderboard

To submit your results to the leaderboard, please add a "predictions" column to [this csv](https://huggingface.co/datasets/mlfoundations/VisIT-Bench/blob/main/visit_bench_single_image.csv), and send to [this mail](mailto:yonatanbitton1@gmail.com).
Please include in your email 1) a name for your model, 2) your team name (including your affiliation), and optionally, 3) a github repo or paper link.
"""
demo = gr.Blocks()
with demo:
    with gr.Row():
        gr.Markdown(headline)

    with gr.Column():
        leaderboard_df = gr.components.DataFrame(
            value=df,
            datatype=["markdown", "markdown", "number", "number", "number"]
        )

demo.launch()