Update app.py

app.py

@@ -76,15 +76,15 @@ model_type = st.radio(
 )
 
 def highlight_obscene_words(text, model_type):
-    label=model_wrapper(text.lower(),model_type)
+    label, prob=model_wrapper(text.lower(),model_type)
     if label=='__label__positive':
         st.markdown(
-            "<span style='background:#47916B;'>{}|приемлемо</span>".format(text), 
+            "<span style='background:#47916B;'>{}:приемлемо</span>".format(text), 
             unsafe_allow_html=True
         )
     else:
         st.markdown(
-            "<span style='background:#ffcccc;'>{}|токсично</span>".format(text), 
+            "<span style='background:#ffcccc;'>{}:токсично-{:.2f}</span>".format(text,prob), 
             unsafe_allow_html=True
         )