limited the number of decimal number outputed

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 import re
+import pandas as pd
+
 # Load the model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("nebiyu29/fintunned-v2-roberta_GA")
 model = AutoModelForSequenceClassification.from_pretrained("nebiyu29/fintunned-v2-roberta_GA")
@@ -104,10 +106,15 @@ st.write("Enter some text, and the model will classify it.")
 text_input = st.text_input("Text Input")
 #if st.button("Classify"):
 predictions = classify(text_input)
+
 labels_str=",".join(predictions["top_labels"])
 probs_ints=",".join(map(str,predictions["top_probabilities"]))
+
+df=pd.DataFrame({'probabilities: ',probs_ints})
+formated_df=df.styled.format("{:.2f}").to_dict('list')
+
     #for prediction in predictions:
        # st.write(f"Segment Text: {prediction['segment_text']}")
         
 st.write(f"Label: {labels_str}")
-st.write(f"Probability: {probs_ints}")
+st.write(f"Probability: {formated_df["probabilities"][0]}")