Spaces:

Cricles
/

demo_detoxi

Sleeping

Cricles commited on Apr 11

Commit

f6e01da

verified ·

1 Parent(s): 4e33d7b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -79,15 +79,14 @@ def highlight_obscene_words(text, model_type):
     if model_type=="FRIDA":
         result=model_wrapper(text, model_type)
         for item in result:
-            st.markdown(f"{item}")
-            if item[1]=="non-toxic":
                 st.markdown(
-                    "<span style='background:#47916B;'>{}:приемлемо</span>".format(item[0]),
                     unsafe_allow_html=True
                 )
             else:
                 st.markdown(
-                    "<span style='background:#ffcccc;'>{}:токсично</span>".format(item[0]),
                     unsafe_allow_html=True
                 )
     else:

     if model_type=="FRIDA":
         result=model_wrapper(text, model_type)
         for item in result:
+            if item[1][0].label=="non-toxic":
                 st.markdown(
+                    "<span style='background:#47916B;'>{}:приемлемо</span>".format(item[1][0].text),
                     unsafe_allow_html=True
                 )
             else:
                 st.markdown(
+                    "<span style='background:#ffcccc;'>{}:токсично</span>".format(item[1][0].text),
                     unsafe_allow_html=True
                 )
     else: