File size: 794 Bytes
4da692a 3df10a0 4da692a 3df10a0 4da692a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | from transformers import pipeline
import gradio as gr
classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
labels = ["service request", "complaint", "general discussion"]
def classify_text(text):
result = classifier(text, labels)
return {label: round(score, 4) for label, score in zip(result["labels"], result["scores"])}
demo = gr.Interface(
fn=classify_text,
inputs=gr.Textbox(lines=4, placeholder="Enter text to classify..."),
outputs=gr.Label(num_top_classes=3), # Display top 3 results
title="Zero-Shot Text Classifier",
description="Classify text into 'service request', 'complaint', or 'general discussion' using Hugging Face's zero-shot classification pipeline.",
)
if __name__ == "__main__":
demo.launch(share=True)
|