Update app.py

app.py

@@ -12,19 +12,21 @@ loaded_model = pickle.load(open("heart_xgb.pkl", 'rb'))
 # Setup SHAP
 explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
 
-gender = {"Female":0,"Male":1}
-chest = {"Typical angina":1,"Atypical angina":2,"Non-anginal pain":3,"Asymptomatic":4}
+gender = {"Female":1,"Male":0}
+chest = {"Typical angina":0,"Atypical angina":1,"Non-anginal pain":2,"Asymptomatic":3}
 choice = {"True":1, "False":0}
-rest = {"Probable or Definite Left Ventricular Hypertrophy by Estes' Criteria":2, "Having ST - T Wave Abnormality":1,"Normal Value":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":2, "Flat":1,"Downsloping":0}
-th = {"Reversible Defect":3, "Normal Blood Flow":2, "Fixed Defect":1}
+sl = {"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':chest[cp],'trtbps':trtbps,'chol':chol, 'fbs':choice[fbs], 'restecg':rest[restecg],
-              'thalachh':thalachh, 'exng':ex[exng], 'oldpeak':oldpeak, 'slp':sl[slp], 'caa':caa, 'thall':th[thall]}, orient = 'index').transpose()
+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,
+              '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)
     
@@ -76,7 +78,7 @@ with gr.Blocks(title=title) as demo:
         with gr.Column():
             age = gr.Slider(label="Age", minimum=0, maximum=100, value=40, step=1)
             sex = gr.Radio(label="Sex", choices =["Female","Male"])
-            cp = gr.Radio(label="Chest Pain Type", choices = ["Typical angina", "Atypical angina", "Non-anginal pain", "Asymptomatic"])
+            cp = gr.Select(label="Chest Pain Type", choices = ["Typical angina", "Atypical angina", "Non-anginal pain", "Asymptomatic"])
             trtbps = gr.Slider(label="Resting Blood Pressure (in mm Hg)", minimum=90, maximum=200, value=90, step=1)
             chol = gr.Slider(label="Cholestoral in mg/dl", minimum=120, maximum=570, value=120, step=1)
             fbs = gr.Radio(label="Fasting Blood Sugar", choices = ["True", "False"])