Update app.py

app.py

@@ -13,10 +13,10 @@ loaded_model = pickle.load(open("h22_xgb.pkl", '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):
-    new_row = pd.DataFrame.from_dict({'ValueDiversity':ValueDiversity,'AdequateResources':AdequateResources,
-              'Voice':Voice,'GrowthAdvancement':GrowthAdvancement,'Workload':Workload,
-              'WorkLifeBalance':WorkLifeBalance}, orient = 'index').transpose()
+def main_func(age,sex,cp,trtbps,chol,fbs,restecg,thalachh,exng,oldpeak,slp,caa,thall,output):
+    new_row = pd.DataFrame.from_dict({'age':age,'sex':sex,'cp':cp,'trtbps':trtbps,'chol':chol,
+              'fbs':fbs,'restecg':restecg,'thalachh':thalachh,'exng':exng,'oldpeak':oldpeak,
+                'slp':slp,'caa':caa,'thall':thall,'output':output}, orient = 'index').transpose()
     
     prob = loaded_model.predict_proba(new_row)