Spaces:

Cricles
/

demo_detoxi

Running

Cricles commited on Apr 11

Commit

9375b5c

verified ·

1 Parent(s): 012ccfd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -77,7 +77,7 @@ model_type = st.radio(
 def highlight_obscene_words(text, model_type):
     if model_type=="FRIDA":
-        result=model_wrapper(text)
         for item in result:
             if item.label=="non-toxic":
                 st.markdown(

 def highlight_obscene_words(text, model_type):
     if model_type=="FRIDA":
+        result=model_wrapper(text, model_type)
         for item in result:
             if item.label=="non-toxic":
                 st.markdown(