Update app.py

app.py

@@ -37,6 +37,10 @@ st.markdown("🔗 **Learn more about employee attrition from [Academy to Innovat
 
 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:")
@@ -91,7 +95,7 @@ def main():
         years_with_curr_manager = st.number_input("Years With Current Manager")
 
     # Predict button
-    if st.button("Predict Employee Attrition 📊"):
+    if st.button("Predict Attrition 📊"):
 
         # Create a DataFrame to hold the user input data
         input_data = pd.DataFrame({