import gradio as gr
from transformers import pipeline

# Load the zero-shot classification pipeline
classifier = pipeline("zero-shot-classification", model="DAMO-NLP-SG/zero-shot-classify-SSTuning-XLM-R")

# Function to perform classification
def classify_text(text, candidate_labels):
    result = classifier(text, candidate_labels)
    return result['labels'][0], result['scores'][0]

# Define the Gradio interface
iface = gr.Interface(
    fn=classify_text,
    inputs=[
        gr.Textbox(label="Enter Text"),
        gr.Textbox(label="Enter Candidate Labels (comma-separated)")
    ],
    outputs=gr.Label(num_top_classes=1),
    live=True,
    title="Zero-Shot Classification Web App",
    description="Enter a text and candidate labels (comma-separated) to classify.",
)

# Launch the Gradio interface
iface.launch(share=True)