Michael Rey
commited on
Commit
·
42f43bd
1
Parent(s):
7e074a9
made changes
Browse files
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}")
|