Spaces:

Cricles
/

demo_detoxi

Sleeping

Cricles commited on Apr 9

Commit

8fba3f9

1 Parent(s): c71fa94

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -57,12 +57,12 @@ def highlight_obscene_words(text):
     label,_=model.predict(text.lower())
     if label[0]=='__label__positive':
         st.markdown(
-            """<span style='background:#47916B;'>|приемлемо</span>""",
             unsafe_allow_html=True
         )
     else:
         st.markdown(
-            """<span style='background:#ffcccc;'>|токсично</span>""",
             unsafe_allow_html=True
         )

     label,_=model.predict(text.lower())
     if label[0]=='__label__positive':
         st.markdown(
+            "<span style='background:#47916B;'>{}|приемлемо</span>".format(text),
             unsafe_allow_html=True
         )
     else:
         st.markdown(
+            "<span style='background:#ffcccc;'>{}|токсично</span>".format(text),
             unsafe_allow_html=True
         )