nurindahpratiwi commited on
Commit
17b69a9
·
1 Parent(s): 14ef77d
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -129,8 +129,10 @@ if st.button("Predict", key="predict_button", help="Click to make a prediction."
129
 
130
  # Make a prediction
131
  prediction = dt_model.predict(final_df)[0]
 
 
132
 
133
 
134
  # Display the prediction
135
- st.write(f"The predicted sales are: {prediction}.")
136
  st.table(input_df)
 
129
 
130
  # Make a prediction
131
  prediction = dt_model.predict(final_df)[0]
132
+ prediction_label = "Beware!!! This customer is likely to Churn" if prediction.item() == "Yes" else "This customer is Not likely churn"
133
+ prediction_label
134
 
135
 
136
  # Display the prediction
137
+ st.write(f"The predicted sales are: {prediction_label}.")
138
  st.table(input_df)