Update app.py

app.py

@@ -296,12 +296,12 @@ if st.button("Process Review"):
             st.write("Pseudo-Sentence:", clean_sentence)
         
         #Display the models prediction
-        if(prediction == 0):
-            st.write("Prediction: Not Spam")
-        elif(prediction == 1):
-            st.write("Prediction: Spam")
+        if prediction == 0:
+            st.markdown("<h3 style='color:green;'>✅ Prediction: Not Spam</h3>", unsafe_allow_html=True)
+        elif prediction == 1:
+            st.markdown("<h3 style='color:red;'>❌ Prediction: Spam</h3>", unsafe_allow_html=True)
         else:
-            st.write("error during prediction")
+            st.markdown("<h3 style='color:orange;'>⚠️ Error during prediction</h3>", unsafe_allow_html=True)
             
     else:
         st.error("Please select a review.")