Spaces:
Sleeping
Sleeping
File size: 544 Bytes
5ccb8df 71945a9 |
1 2 3 4 5 6 7 8 9 10 11 |
from transformers import pipeline
classifier = pipeline("zero-shot-classification",
model="facebook/bart-large-mnli")
import gradio as gr
def classifier(sequence):
candidate_labels = ['travel', 'cooking', 'dancing']
results = classifier(sequence, candidate_labels)
return {label: score for label, score in zip(results['labels'], results['scores'])}
demo_classifier = gr.Interface(fn = classifier, title="Text_classifier", inputs= gr.Textbox(), outputs=gr.Label(num_top_classes=3))
demo_classifier.launch() |