Update app.py

app.py

@@ -12,7 +12,7 @@ loaded_model = pickle.load(open("heart_xgb.pkl", 'rb'))
 # Setup SHAP
 explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
 
-sex = {"Female":1,"Male":0}
+gender = {"Female":1,"Male":0}
 cp = {"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}
@@ -22,7 +22,7 @@ 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':[sex],
+    new_row = pd.DataFrame.from_dict({'age':age,'sex':gender[sex],
               'cp':cp[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]},