|
|
from transformers import pipeline |
|
|
import gradio as grad |
|
|
zero_shot_classifier = pipeline("zero-shot-classification") |
|
|
|
|
|
|
|
|
def classify(text,labels): |
|
|
classifer_labels = labels.split(",") |
|
|
|
|
|
response = zero_shot_classifier(text,classifer_labels) |
|
|
return response |
|
|
txt=grad.Textbox(lines=1, label="English", placeholder="text to be classified") |
|
|
labels=grad.Textbox(lines=1, label="Labels", placeholder="comma separated labels") |
|
|
out=grad.Textbox(lines=1, label="Classification") |
|
|
grad.Interface(classify, inputs=[txt,labels], outputs=out).launch() |
|
|
|
