Michael Rey commited on
Commit
42f43bd
·
1 Parent(s): 7e074a9

made changes

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -69,4 +69,5 @@ if st.button("Predict Purchase"):
69
 
70
  st.subheader("Prediction Result")
71
  result_text = "Yes! The user is likely to purchase." if prediction == 1 else "No, the user is not likely to purchase."
 
72
  st.write(f"Confidence: {prediction_proba[prediction]:.2f}")
 
69
 
70
  st.subheader("Prediction Result")
71
  result_text = "Yes! The user is likely to purchase." if prediction == 1 else "No, the user is not likely to purchase."
72
+ st.success(result_text) if prediction == 1 else st.warning(result_text)
73
  st.write(f"Confidence: {prediction_proba[prediction]:.2f}")