Update app.py

app.py

@@ -80,8 +80,6 @@ if submit:
 
             prediction = np.argmax(result, axis=-1)
             if prediction == 0:
-                st.success(f"This news is {label[prediction]}.")
-            else:
                 st.markdown(f"""<p style="background-color: rgb(254, 242, 242); 
                 color: rgb(153, 27, 27);
                 font-size: 20px;
@@ -91,6 +89,16 @@ if submit:
                 padding-bottom: 15px;
                 line-height: 25px;
                 text-align: center;">This article is <b>{label[prediction]}</b>.</p>""", unsafe_allow_html=True)
+            else:
+                st.markdown(f"""<p style="background-color: rgb(236, 253, 245); 
+                color: rgb(6, 95, 70);
+                font-size: 20px;
+                border-radius: 7px;
+                padding-left: 12px;
+                padding-top: 15px;
+                padding-bottom: 15px;
+                line-height: 25px;
+                text-align: center;">This article is <b>{label[prediction]}</b>.</p>""", unsafe_allow_html=True)
             
             with st.expander("Related Articles"):
                 for i in sorted[:5]: