Update app.py

app.py

@@ -56,9 +56,7 @@ st.write("""<p style='text-align: center; font-size: 24px;'>Это прилож
 
 def highlight_obscene_words(text):
     label,probs=model.predict(text)
-    st.write(f"{label}")
-    st.write(f"{probs}")
-    if label=='__label__positive':
+    if label[0]=='__label__positive':
         return (text + " ", "приемлемо", "#47916B")
     else:
         return (text + " ", "токсично", "#ffcccc")