Update app.py

app.py

@@ -1,31 +1,20 @@
 import gradio as gr
-
 from transformers import pipeline
 
-
+classifier = pipeline("zero-shot-classification")
 
 def classify(text):
-
-    classifier = pipeline("zero-shot-classification")
-
     candidate_labels = ["positive", "negative", "neutral"]
-
     output = classifier(text, candidate_labels)
-
-    return output
-
-
-
-    
+    # Process the output to match Gradio's expected input format for gr.Label
+    labels = output['labels']
+    scores = output['scores']
+    # Construct a simple string representation of top classifications
+    top_classes = ', '.join([f"{labels[i]}: {scores[i]:.2f}" for i in range(len(labels))])
+    return top_classes
 
 demo = gr.Interface(fn=classify,
+                    inputs=gr.Textbox(label="Enter something"),
+                    outputs=gr.Label())
 
-                   inputs=gr.Textbox(label="Enter something"),
-
-                   outputs=gr.Label(num_top_classes=3))
-
-
-
-
-
-demo.launch()
+demo.launch()