import gradio as gr
from transformers import pipeline

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

def classify(text, labels):
    return classifier(text, candidate_labels=labels)

iface = gr.Interface(
    fn=classify,
    inputs=[gr.Text(label="Text"), gr.Textbox(lines=1, label="Candidate Labels (comma-separated)")],
    outputs="json"
)

if __name__ == "__main__":
    iface.launch()