Update app.py

app.py

@@ -5,20 +5,22 @@ from transformers import pipeline
 pipeline = pipeline("text-classification", model="ProsusAI/finbert", trust_remote_code=True)
 
 def predict(input_text):
-    predictions = pipeline(input_text, threshold=0.5, return_scores=True)
-    return predictions[0]
+  predictions = pipeline(input_text, threshold=0.5, return_scores=True)
+  return predictions[0]
 
+# Define the Gradio interface
 gradio_app = gr.Interface(
-    predict,
-    inputs=gr.Textbox(label="Write a text"),
-    outputs=None,
-    components=[
-        gr.Label(label="Label: {}".format(predictions[0]["label"])),
-        gr.Label(label="Score: {:.2f}".format(predictions[0]["score"][0])),
-    ],
-    title="Financial Sentiment Analysis",
+  predict,
+  inputs=gr.Textbox(label="Write a text"),
+  outputs=None, # Remove outputs here
+  components=[ # Add components to display probabilities
+    gr.Label(label="Neutral: {:.2f}".format(predictions[0]["score"][0])),
+    gr.Label(label="Positive: {:.2f}".format(predictions[0]["score"][1])),
+    gr.Label(label="Negative: {:.2f}".format(predictions[0]["score"][2])),
+  ],
+  title="Financial Sentiment Analysis",
 )
 
 # Launch the Gradio interface
 if __name__ == "__main__":
-    gradio_app.launch()
+  gradio_app.launch()