taxfree_python
Add functions to submit models
b29fd2d
raw
history blame
1.24 kB
import gradio as gr
from leaderboard.dataset import get_leaderboard_df
from leaderboard.submission import submit_model
def display_leaderboard():
df = get_leaderboard_df()
return df
with gr.Blocks() as app:
gr.Markdown("# human_methylation_bench_ver1 Leaderboard")
with gr.Tab("Leaderboard"):
leaderboard_df = gr.DataFrame(
value=display_leaderboard(),
headers=["Model Name", "Score (relative_error_loss)", "Rank"],
interactive=False,
label="Leaderboard",
)
with gr.Tab("Submit Model"):
model_name_input = gr.Textbox(label="Model Name", placeholder="e.g. My Great Model")
model_url_input = gr.Textbox(
label="Hugging Face Model Joblib URL",
placeholder="e.g. https://huggingface.co/username/model/resolve/main/model.joblib",
)
submit_button = gr.Button("Submit")
submission_output = gr.DataFrame(
headers=["Model Name", "Score (relative_error_loss)", "Rank"], interactive=False, label="Updated Leaderboard"
)
submit_button.click(submit_model, inputs=[model_name_input, model_url_input], outputs=submission_output)
if __name__ == "__main__":
app.launch()