Show all labels

app.py

@@ -11,14 +11,14 @@ pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer)
 
 # Define the classify_text function using the pipeline
 def classify_text(text):
-    prediction = pipe(text)[0]["label"]
+    prediction = pipe(text)[0]
     return prediction
 
 # Create a Gradio interface
 iface = gr.Interface(
     fn=classify_text,
     inputs=gradio.components.Textbox(label="Enter text"),
-    outputs=gr.outputs.Label(label="Predicted classes"),
+    outputs=gr.outputs.Label(num_top_classes=None),  # Show all labels
 )
 
 # Launch the Gradio app