ZeroShot / app.py
AudreyMireille's picture
Create app.py
5ccb8df verified
raw
history blame
No virus
382 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'])}