Add survival probability

app.py

@@ -41,6 +41,7 @@ passenger_data = {
 passenger_data_df = pd.DataFrame.from_dict([passenger_data])
 
 prediction = model.predict(passenger_data_df)[0]
+prediction_proba = model.predict_proba(passenger_data_df)[:,1][0]
 
 if prediction == 0:
     st.markdown("# YOU'RE DEAD :cry:")
@@ -49,4 +50,6 @@ elif sex == "Male":
     st.markdown("# YOU MADE IT, :man-swimming:")
 
 else:
-    st.markdown("# YOU MADE IT, :woman-swimming:")
+    st.markdown("# YOU MADE IT, :woman-swimming:")
+    
+st.write(f"Probabilty of survival {round(prediction_proba, 2)* 100}%.")