Update app.py

app.py

@@ -35,7 +35,9 @@ st.write("Enter the email content below:")
 
 user_input = st.text_area("Email Content")
 
-if st.button("Classify"):
-    result = predict_sentence(user_input, wordvect_model, classifier_model)
-    st.write(f"The email is classified as: {result}")
+result = predict_sentence(user_input, wordvect_model, classifier_model)
+    if result == "spam":
+        st.markdown(f"<p style='color:red; font-size:20px; font-weight:bold;'>The email is classified as: {result}</p>", unsafe_allow_html=True)
+    else:
+        st.markdown(f"<p style='color:green; font-size:20px; font-weight:bold;'>The email is classified as: {result}</p>", unsafe_allow_html=True)