Update app.py

app.py

@@ -34,19 +34,14 @@ st.image("https://www.aihr.com/wp-content/uploads/Reasons-for-Employee-Attrition
 
 # Link to Detailed Article on Employee Attrition
 st.markdown("🔗 **Learn more about employee attrition from [Academy to Innovate HR (AIHR)](https://www.aihr.com/wp-content/uploads/Reasons-for-Employee-Attrition.png)**")
-
 st.markdown("---")
 
-# Display an icon representing employee attrition
-st.markdown('<i class="fas fa-user-slash"></i> Employee Attrition', unsafe_allow_html=True)
-
-
 # Additional Information for Sample Prediction
 st.write("📊 To make a prediction, input the information of the employee whose attrition you want to predict.")
 st.write("Please provide the following information to make a prediction:")
 
 # About Section with Style
-st.sidebar.title("ℹ️ About")
+st.sidebar.title("About")
 st.sidebar.info(
     "This app predicts employee attrition using machine learning on HR data, aiding HR professionals in retention strategies. "
     "It utilizes a CatBoost machine learning model trained on an employee attrition dataset."
@@ -124,6 +119,13 @@ def main():
         # Make predictions
         prediction = model.predict(input_data)
         probability = model.predict_proba(input_data)[:, 1]
+
+        # Display predicted attrition status
+        st.subheader("Predicted Attrition Status 🔮")
+        if prediction[0] == 1:
+            st.error("Employee is predicted to leave (Attrition = Yes)")
+        else:
+            st.success("Employee is predicted to stay (Attrition = No)")
         
         # Display prediction probability
         if prediction[0] == 1: