import gradio as gr
from transformers import pipeline

model = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")


def predict(field_name, property_list):
    properties = property_list.split(",")
    results = model(field_name, properties)
    top_results = []
    for i, result in enumerate(results["labels"]):
        if i < 10:
            top_results.append(f"{result}: {results['scores'][i]}")
    return results["labels"][0], "\n".join(top_results)


demo = gr.Interface(
    fn=predict, inputs=["text", "textarea"], outputs=["text", "textarea"]
)

demo.launch()