Update app.py | Remove confidence

app.py

@@ -16,18 +16,16 @@ def predict_sentiment(text):
     predicted_class = torch.argmax(predictions, dim=1).item()
     
     sentiment_map = {0: "NEGATIVE", 1: "NEUTRAL", 2: "POSITIVE"}
-    confidence = predictions[0][predicted_class].item()
     
     # Format each output separately
     sentiment_output = f"{sentiment_map[predicted_class]}"
-    confidence_output = f"{confidence:.3f}"
     probabilities_output = (
         f"Negative: {predictions[0][0]:.3f}\n"
         f"Neutral: {predictions[0][1]:.3f}\n"
         f"Positive: {predictions[0][2]:.3f}"
     )
 
-    return sentiment_output, confidence_output, probabilities_output
+    return sentiment_output, probabilities_output
 
 # Define Gradio Interface with 3 separate outputs
 with gr.Blocks(theme="soft") as demo:
@@ -58,12 +56,6 @@ with gr.Blocks(theme="soft") as demo:
                 max_lines=5,
                 interactive=False
             )
-            confidence_box = gr.Textbox(
-                label="Confidence",
-                lines=3,
-                max_lines=5,
-                interactive=False
-            )
             probabilities_box = gr.Textbox(
                 label="Raw Probabilities",
                 lines=6,
@@ -75,13 +67,13 @@ with gr.Blocks(theme="soft") as demo:
     submit_btn.click(
         fn=predict_sentiment,
         inputs=input_box,
-        outputs=[sentiment_box, confidence_box, probabilities_box]
+        outputs=[sentiment_box, probabilities_box]
     )
 
     clear_btn.click(
         fn=lambda: ("", "", ""),
         inputs=None,
-        outputs=[input_box, sentiment_box, confidence_box, probabilities_box]
+        outputs=[input_box, sentiment_box, probabilities_box]
     )
 
     # Add examples below
@@ -94,7 +86,7 @@ with gr.Blocks(theme="soft") as demo:
             ["Ganda legit, kumpleto... problema lang nainit ng sobra..."]
         ],
         inputs=input_box,
-        outputs=[sentiment_box, confidence_box, probabilities_box],
+        outputs=[sentiment_box, probabilities_box],
         label="Example Sentences"
     )