Lord-Raven
Trying zero-shot classification.
6deae7c
raw
history blame
499 Bytes
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()