Update app.py

app.py

@@ -84,9 +84,14 @@ def main():
         # Make predictions
         prediction = model.predict(input_data)
         probability = model.predict_proba(input_data)[:, 1]
+
+        # Display prediction probability
+        if prediction[0] == 1:
+            st.subheader("Prediction Probability")
+            st.write(f"The probability of the employee leaving is: {probability[0]*100:.2f}%")
         
         # Display characteristic-based recommendations
-        st.subheader("Suggestions for retaining the employee:")
+        st.subheader("Recommendations for Retaining The Employee:")
         if job_satisfaction == 1 or environment_satisfaction == 1:
             st.markdown("- Enhance job and environment satisfaction through initiatives such as recognition programs and improving workplace conditions.")
         if years_since_last_promotion > 5: