Spaces:
Runtime error
Runtime error
import gradio | |
import gradio as gr | |
from transformers import pipeline | |
classifier = pipeline("zero-shot-classification", model="MoritzLaurer/deberta-v3-large-zeroshot-v1") | |
def classify(review, candidate_labels=('positive', 'neutral', 'negative')): | |
labels = classifier(review, candidate_labels) | |
return gradio.Label(label='Classification', value=(dict(zip(labels['labels'], labels['scores'])))) | |
app = gr.Interface(fn=classify, inputs=['text'], outputs=['label'], allow_flagging="never") | |
app.launch() | |