Update app.py

app.py

@@ -60,7 +60,9 @@ def predict_employee_status(satisfaction_level, last_evaluation, number_project,
     try:
         dmatrix = xgb.DMatrix(input_df)
         prediction = model.predict(dmatrix)
-        return "✅ Employee is likely to quit." if prediction[0] > 0.5 else "✅ Employee is likely to stay."
+        prediction_prob = prediction[0]
+        result = "✅ Employee is likely to quit." if prediction_prob > 0.5 else "✅ Employee is likely to stay."
+        return f"{result} (Probability: {prediction_prob:.2%})"
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
@@ -84,7 +86,8 @@ interface = gr.Interface(
     ],
     outputs="text",
     title="Employee Retention Prediction System",
-    description="Predict whether an employee is likely to stay or quit based on their profile."
+    description="Predict whether an employee is likely to stay or quit based on their profile.",
+    theme="dark"
 )
 
 interface.launch()