Spaces:
Sleeping
Sleeping
File size: 561 Bytes
514367c c9206a9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
from transformers import pipeline
import gradio as grad
zero_shot_cls = pipeline("zero-shot-classification")
def classify(text, labels):
cls_labels = labels.split(",")
#["IT", "software", "marketing", "sales", "R&D", "logistics"]
response = zero_shot_cls(text, cls_labels)
return response
in_text = grad.Textbox(lines=1, label="English", placeholder="Text to be classified")
in_labels = grad.Textbox(lines=1, label="Labels", placeholder="Comma separated labels")
out = grad.Textbox(lines=1, inputs=[in_text, in_labels], outputs=out).launch() |