Update app.py

app.py

@@ -2,7 +2,7 @@ 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")
+classifier = pipeline("zero-shot-classification")
 
 # Function to perform classification
 def classify_text(text, candidate_labels):
@@ -14,9 +14,8 @@ 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),
+        gr.Textbox(label="Enter Candidate Labels (comma-separated)"),
+    outputs=gr.Textbox(),
     live=True,
     title="Zero-Shot Classification Web App",
     description="Enter a text and candidate labels (comma-separated) to classify.",