Lord-Raven
Trying zero-shot classification.
47a0109
raw
history blame
541 Bytes
import gradio
from transformers import pipeline
def zero_shot_classification(data_string):
data = json.loads(data_string)
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 = "text",
outputs = "json"
)
gradio_interface.launch()