Update app.py

app.py

@@ -13,7 +13,7 @@ loaded_model = pickle.load(open("XGBclf_v0.2.sav", 'rb'))
 explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
 
 # Create the main function for server
-def main_func(ValueDiversity,AdequateResources,Voice,GrowthAdvancement,Workload,WorkLifeBalance):
+def main_func(Management,EmployeeWellBeing,EngagedAtWork,WorkEnvironment,JobSatisfaction):
     new_row = pd.DataFrame.from_dict({'Management':Management,'EmployeeWellBeing':EmployeeWellBeing,
               'EngagedAtWork':EngagedAtWork,'WorkEnvironment':WorkEnvironment,'JobSatisfaction':JobSatisfaction,
               }, orient = 'index').transpose()
@@ -62,7 +62,7 @@ with gr.Blocks(title=title) as demo:
         
             submit_btn.click(
                 main_func,
-                [ValueDiversity,AdequateResources,Voice,GrowthAdvancement,Workload,WorkLifeBalance],
+                [Management,EmployeeWellBeing,EngagedAtWork,WorkEnvironment,JobSatisfaction],
                 [label,local_plot], api_name="Employee_Turnover"
             )