Spaces:

Cricles
/

demo_detoxi

Sleeping

Cricles commited on Apr 9

Commit

f2f78a0

1 Parent(s): 9077599

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -54,7 +54,7 @@ st.write("""<p style='text-align: center; font-size: 24px;'>Это прилож
 И даже не придётся платить 300 bucks.</p>""", unsafe_allow_html=True)
 def highlight_obscene_words(text):
-    label,probs=model.predict(text)
     if label[0]=='__label__positive':
         st.markdown(
             "<span style='background:#47916B;'>приемлемо</span>",

 И даже не придётся платить 300 bucks.</p>""", unsafe_allow_html=True)
 def highlight_obscene_words(text):
+    label,probs=model.predict(text.lower())
     if label[0]=='__label__positive':
         st.markdown(
             "<span style='background:#47916B;'>приемлемо</span>",