Update app.py

app.py

@@ -93,11 +93,14 @@ def main():
         prediction = model.predict(input_data)
         probability = model.predict_proba(input_data)[:, 1]
 
-        # Display prediction
+        # Display prediction and probability
         if prediction[0] == 0:
             st.success("Employee is predicted to stay (Attrition = No)")
         else:
             st.error("Employee is predicted to leave (Attrition = Yes)")
+            # Show prediction probability
+            st.subheader("Prediction Probability")
+            st.write(f"The probability of the employee leaving is: {probability[0]*100:.2f}%")
 
             # Offer recommendations for retaining the employee
             st.subheader("Suggestions for retaining the employee:")
@@ -109,8 +112,5 @@ def main():
             st.markdown("- Introduce or enhance policies that support work-life balance for employees with families.")
             st.markdown("- Recognize the unique challenges and opportunities within each department and tailor retention strategies accordingly.")
 
-        # Display probability
-        st.write(f"Probability of Attrition: {probability[0]*100:.2f}%")
-
 if __name__ == "__main__":
     main()