Small changes to improve the app (gauge for probability)

app.py

@@ -4,6 +4,7 @@ import os
 import matplotlib.pyplot as plt
 from sentence_transformers import SentenceTransformer
 from tensorflow.keras.models import model_from_json
+import plotly.graph_objects as go
 from PIL import Image
 import streamlit as st
 
@@ -193,7 +194,25 @@ with st.expander('Input a sentence and check the probability of it being hateful
     y_pred = loaded_model.predict(preprocessed_sentence)
     percentage = y_pred[0][0] * 100
 
-    st.write(f'Probability of being hateful: {round(percentage, 2)/100}')
+    # Scoreboard du prêt
+    col1, col2, col3, col4 = st.columns(4)
+
+    with col2:
+
+        fig1 = go.Figure(go.Indicator(
+            mode = "gauge+number+delta",
+            value = y_pred[0][0],
+            delta = {'reference': 0.50},
+            gauge = {'axis': {'range': [None, 1]},
+                    'steps' : [
+                        {'range': [0, 0.5], 'color': "lightgray"},
+                        {'range': [0.5, 0.75], 'color': "gray"}],
+                    'threshold' : {'line': {'color': "red", 'width': 4}, 'thickness': 0.9, 'value': 0.5}},
+            domain = {'x': [0, 1], 'y': [0, 1]},
+            title = {'text': 'Probability of your sentence being hateful'}))
+
+        st.plotly_chart(fig1)
+
     if y_pred[0][0] < 0.5:
         st.write(f"Congrats, it's not hateful!!!")
     else:

requirements.txt

@@ -4,4 +4,5 @@ numpy
 matplotlib
 sentence_transformers
 tensorflow
-Pillow
+Pillow
+plotly