import gradio from transformers import pipeline def zero_shot_classification(data_string): print(data_string) data = json.loads(data_string) print(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 = gradio.Textbox(label="JSON Input"), outputs = "json" ) gradio_interface.launch()