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() |