danieldux's picture
Add Gradio app for calculating hierarchical precision and recall
f78d989
raw
history blame
965 Bytes
# filename: gradioapp1.py
import gradio as gr
from ham import calculate_hierarchical_precision_recall
from isco import create_hierarchy_dict
# from tests import test_cases
isco_csv_url = (
"https://www.ilo.org/ilostat-files/ISCO/newdocs-08-2021/ISCO-08/ISCO-08%20EN.csv"
)
hierachy = create_hierarchy_dict(isco_csv_url)
# Define the Gradio interface
iface = gr.Interface(
fn=calculate_hierarchical_precision_recall,
inputs=[
gr.Textbox(lines=2, placeholder="Enter true labels separated by commas"),
gr.Textbox(lines=2, placeholder="Enter predicted labels separated by commas"),
gr.Textbox(lines=5, placeholder="Enter hierarchy in JSON format"),
],
outputs="textbox",
examples=[
# Add example inputs here if necessary
],
description="This app calculates hierarchical precision and recall. Enter the true labels, predicted labels, and the hierarchy in JSON format.",
)
# Run the app
iface.launch()