Update app.py

app.py

@@ -13,19 +13,19 @@ loaded_model = pickle.load(open("heart_xgb.pkl", 'rb'))
 explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
 
 gender = {"Female":1,"Male":0}
-cp = {"Typical angina":0,"Atypical angina":1,"Non-anginal pain":2,"Asymptomatic":3}
+chestpain = {"Typical angina":0,"Atypical angina":1,"Non-anginal pain":2,"Asymptomatic":3}
 bloodsuguar = {"True":1, "False":0}
 rest = {"Probable or Definite Left Ventricular Hypertrophy by Estes' Criteria":0, "Having ST - T Wave Abnormality":1,"Normal Value":2}
 ex = {"Yes":1, "No":0}
-sl = {"Upsloping":0, "Flat":1,"Downsloping":2}
+slope = {"Upsloping":0, "Flat":1,"Downsloping":2}
 th = {"Reversible Defect":0, "Normal Blood Flow":1, "Fixed Defect":2}
 
 # 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({'age':age,'sex':gender[sex],
-              'cp':cp[cp],'trtbps':trtbps,'chol':chol,
+              'cp':chestpain[cp],'trtbps':trtbps,'chol':chol,
               'fbs':bloodsuguar[fbs], 'restecg':rest[restecg],'thalachh':thalachh,'exng':ex[exng],
-                                     'oldpeak':oldpeak,'slp':sl[slp],'caa':caa,'thall':thl[thall]}, 
+                                     'oldpeak':oldpeak,'slp':slope[slp],'caa':caa,'thall':thl[thall]}, 
                                      orient = 'index').transpose()
     
     prob = loaded_model.predict_proba(new_row)