Small improvement in the app (gauge for probability)

5_artefact_streamlit.py

@@ -1,8 +1,10 @@
 import pandas as pd
 import numpy as np
+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
 
@@ -192,7 +194,21 @@ 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
+    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
+plotly
 Pillow