Update app.py

app.py

@@ -199,7 +199,7 @@ if submitted:
     # Making Predictions
     # Passing data to pipeline to make prediction
     pred_output = pipeline_of_my_model.predict(df)
-    prob_output = np.max(pipeline_of_my_model.predict_proba(df))
+    prediction_proba = pipeline_of_my_model.predict_proba(df)
     
     # Interpleting prediction output for display
     X= pred_output[-1]
@@ -214,12 +214,11 @@ if submitted:
     st.markdown('''---''')
     st.markdown("<h4 style='text-align: center;'> Prediction Results </h4> ", unsafe_allow_html=True)
     st.success(f"Prediction: {output}")
-    #st.success(f"Confidence Probability: {prob_output}")
     st.markdown('''---''')
 
     st.subheader('Prediction Percentages:') 
-    st.write('**Probablity of Person not churning is ( in % )**:',prob_output[0][0]*100)
-    st.write('**Probablity of Person not churning is ( in % )**:',prob_output[0][1]*100)
+    st.write('**Probablity of Person not churning is ( in % )**:',prediction_proba[0][0]*100)
+    st.write('**Probablity of Person not churning is ( in % )**:',prediction_proba[0][1]*100)
     
 
     # Making expander to view all records