wszhang1021 commited on
Commit
135c0af
·
1 Parent(s): 265506c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ loaded_model = pickle.load(open("heart.xgb.pkl", 'rb'))
13
  explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
14
 
15
  # Create the main function for server
16
- def main_func(ValueDiversity,AdequateResources,Voice,GrowthAdvancement,Workload,WorkLifeBalance):
17
  new_row = pd.DataFrame.from_dict({'ValueDiversity':ValueDiversity,'AdequateResources':AdequateResources,
18
  'Voice':Voice,'GrowthAdvancement':GrowthAdvancement,'Workload':Workload,
19
  'WorkLifeBalance':WorkLifeBalance}, orient = 'index').transpose()
 
13
  explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
14
 
15
  # Create the main function for server
16
+ def main_func(age, sex, cp, trtbps, chol, fbs, restecg, thalachh, exng, oldpeak, slp, caa, thall):
17
  new_row = pd.DataFrame.from_dict({'ValueDiversity':ValueDiversity,'AdequateResources':AdequateResources,
18
  'Voice':Voice,'GrowthAdvancement':GrowthAdvancement,'Workload':Workload,
19
  'WorkLifeBalance':WorkLifeBalance}, orient = 'index').transpose()