Update app.py

app.py

@@ -39,7 +39,7 @@ def main_func(SupportiveGM, Merit, LearningDevelopment, WorkEnvironment, Engagem
     plt.rcParams['figure.figsize'] = (6, 4)
     plt.close()
 
-    return {"Leave": 1 - float(prob[0]), "Stay": float(prob[0])}, local_plot
+   return {"Leave": float(prob[0][0]), "Stay": 1-float(prob[0][0])}, local_plot
 
 # Create the UI
 title = "**Mod 3 Team 5: Employee Turnover Predictor**"