Update app.py

app.py

@@ -14,9 +14,10 @@ explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
 
 # Create the main function for server
 def main_func(age, sex, cp, trtbps, chol, fbs, restecg, thalachh, exng, oldpeak, slp, caa, thall):
-    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({'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},
+              orient = 'index').transpose()
     
     prob = loaded_model.predict_proba(new_row)