import gradio from transformers import pipeline def zero_shot_classification(data): classifier = pipeline('zero-shot-classification, model='Xenova/mobilebert-uncased-mnli') results = classifier(data.sequence, candidate_labels=data.candidate_labels, hypothesis_template=data.hypothesis_template, multi_label=data.multi_label) return {results} gradio_interface = gradio.Interface( fn = zero_shot_classification, inputs = "json", outputs = "json" ) gradio_interface.launch()