File size: 518 Bytes
31a7455 5d00e2d 31a7455 656c4ee 5d00e2d 18418b8 5d00e2d c7102bb 72eacc1 31a7455 72eacc1 31a7455 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import gradio as gr
from transformers import pipeline
def do_action(text):
classifier = pipeline("zero-shot-classification",
model="facebook/bart-large-mnli")
candidate_labels = ['gun control', 'abortion', 'gender transition', 'freedom of speech', 'immigration', 'taxes']
result = classifier(text, candidate_labels)
result = dict(zip(result['labels'], result['scores']))
return result
iface = gr.Interface(fn=do_action, inputs="text", outputs="label")
iface.launch()
|