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