Update app.py

app.py

@@ -150,7 +150,7 @@ if st.button("Predict"):
                     # Display the prediction using st.write
                     st.write(f"Prediction for the selected URL ({selected_url}): **{prediction[0]}**")
                     # Display the prediction result
-                    if prediction[0] == 0:
+                    if prediction[0] == 0 or prediction[0] == "legitimate":
                         st.success("The website is not a phishing website.")
                         st.markdown(f'<img src="data:image/gif;base64,{data_url_ok}" alt="cat gif">', unsafe_allow_html=True,)
                     else: