Spaces:
Build error
Build error
Commit ·
135c0af
1
Parent(s): 265506c
Update app.py
Browse files
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(
|
| 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()
|