Update app.py

app.py

@@ -13,9 +13,9 @@ loaded_model = pickle.load(open("XGB_CLF_5Feature.pkl", 'rb'))
 explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
 
 # Create the main function for server
-def main_func(Management,EmployeeWellBeing,EngagedAtWork,WorkEnvironment,JobSatisfaction):
-    new_row = pd.DataFrame.from_dict({'Management':Management,'EmployeeWellBeing':EmployeeWellBeing,
-              'EngagedAtWork':EngagedAtWork,'WorkEnvironment':WorkEnvironment,'JobSatisfaction':JobSatisfaction,
+def main_func(JobSatisfaction,EmployeeWellBeing,EngagedAtWork,WorkEnvironment,Management):
+    new_row = pd.DataFrame.from_dict({'JobSatisfaction':JobSatisfaction,'EmployeeWellBeing':EmployeeWellBeing,
+              'EngagedAtWork':EngagedAtWork,'WorkEnvironment':WorkEnvironment,'Management':Management,
               }, orient = 'index').transpose()
     
     prob = loaded_model.predict_proba(new_row)
@@ -50,11 +50,11 @@ with gr.Blocks(title=title) as demo:
     gr.Markdown("""---""")
     with gr.Row():        
         with gr.Column():
-            Management = gr.Slider(label="Management Score", minimum=1, maximum=5, value=4, step=.1)
+            JobSatisfaction = gr.Slider(label="Job Satisfaction Score", minimum=1, maximum=5, value=4, step=.1)
             EmployeeWellBeing = gr.Slider(label="Employee Well Being Score", minimum=1, maximum=5, value=4, step=.1)
             EngagedAtWork = gr.Slider(label="Work Engagement Score", minimum=1, maximum=5, value=4, step=.1)
             WorkEnvironment = gr.Slider(label="Work Environment Score", minimum=1, maximum=5, value=4, step=.1)
-            JobSatisfaction = gr.Slider(label="Job Satisfaction Score", minimum=1, maximum=5, value=4, step=.1)
+            Management = gr.Slider(label="Management Score", minimum=1, maximum=5, value=4, step=.1)
             submit_btn = gr.Button("Analyze")
         with gr.Column(visible=True) as output_col:
             label = gr.Label(label = "Predicted Label")
@@ -62,13 +62,13 @@ with gr.Blocks(title=title) as demo:
         
             submit_btn.click(
                 main_func,
-                [Management,EmployeeWellBeing,EngagedAtWork,WorkEnvironment,JobSatisfaction],
+                [JobSatisfaction,EmployeeWellBeing,EngagedAtWork,WorkEnvironment,Management],
                 [label,local_plot], api_name="Employee_Turnover"
             )
     
     gr.Markdown("### Click on any of the examples below to see how it works:")
     gr.Examples([[4,4,4,4,5,5], [5,4,5,4,4,4]], 
-                [Management,EmployeeWellBeing,EngagedAtWork,WorkEnvironment,JobSatisfaction], 
+                [JobSatisfaction,EmployeeWellBeing,EngagedAtWork,WorkEnvironment,Management], 
                 [label,local_plot], main_func, cache_examples=True)
 
 demo.launch()