Spaces:
Running
on
Zero
Running
on
Zero
| 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() |