Update app.py

app.py

@@ -14,9 +14,9 @@ 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):
-    new_row = pd.DataFrame.from_dict({'ValueDiversity':ValueDiversity,'AdequateResources':AdequateResources,
-              'Voice':Voice,'GrowthAdvancement':GrowthAdvancement,'Workload':Workload,
-              'WorkLifeBalance':WorkLifeBalance}, orient = 'index').transpose()
+    new_row = pd.DataFrame.from_dict({'Management':Management,'EmployeeWellBeing':EmployeeWellBeing,
+              'EngagedAtWork':EngagedAtWork,'WorkEnvironment':WorkEnvironment,'JobSatisfaction':JobSatisfaction,
+              }, orient = 'index').transpose()
     
     prob = loaded_model.predict_proba(new_row)