Update app.py

app.py

@@ -4,15 +4,24 @@ from transformers import pipeline
 # Load the zero-shot classification pipeline
 classifier = pipeline("zero-shot-classification")
 
+# 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=lambda text, candidate_labels: classifier(text, candidate_labels),
-    inputs=[gr.Textbox(label="Enter Text"), gr.Textbox(label="Enter Candidate Labels (comma-separated)")],
-    outputs=gr.Label(),
+    fn=classify_text,
+    inputs=[
+        gr.Textbox(label="Enter Text"),
+        gr.Textbox(label="Enter Candidate Labels (comma-separated)"),
+        gr.Button(text="Submit")
+    ],
+    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)
+iface.launch(share=True)