Update pages/🤢 You_are_toxic.py

pages/🤢 You_are_toxic.py

@@ -103,17 +103,17 @@ outputs = model_bert(**inputs)
 prediction_bert = torch.nn.functional.softmax(outputs.logits, dim=1)
 prediction_bert = torch.argmax(prediction_bert, dim=1).numpy()
 st.write("Comment by ML model:", user_review)
-
-if prediction == 0:
-    st.markdown("<p style='color: green;'>Non-toxic comment</p>", unsafe_allow_html=True)
-else:
-    st.markdown("<p style='color: red;'>Toxic comment</p>", unsafe_allow_html=True)
-st.write("Comment by RuBERT:", user_review)
-
-if prediction_bert == 0:
-    st.markdown("<p style='color: green;'>Controversial comment</p>", unsafe_allow_html=True)
-elif prediction_bert == 1:
-    st.markdown("<p style='color: red;'>Non-toxic comment</p>", unsafe_allow_html=True)
-else:
-    st.markdown("<p style='color: red;'>Toxic comment</p>", unsafe_allow_html=True)
+if user_review.strip():
+    if prediction == 0:
+        st.markdown("<p style='color: green;'>Non-toxic comment</p>", unsafe_allow_html=True)
+    else:
+        st.markdown("<p style='color: red;'>Toxic comment</p>", unsafe_allow_html=True)
+    st.write("Comment by RuBERT:", user_review)
+    
+    if prediction_bert == 0:
+        st.markdown("<p style='color: green;'>Controversial comment</p>", unsafe_allow_html=True)
+    elif prediction_bert == 1:
+        st.markdown("<p style='color: red;'>Non-toxic comment</p>", unsafe_allow_html=True)
+    else:
+        st.markdown("<p style='color: red;'>Toxic comment</p>", unsafe_allow_html=True)