ZeroShot / app.py
AudreyMireille's picture
Update app.py
71945a9 verified
raw
history blame contribute delete
No virus
544 Bytes
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()