Update app.py

app.py

@@ -7,20 +7,28 @@ import numpy as np
 import matplotlib.pyplot as plt
 
 # load the model from disk
-loaded_model = pickle.load(open("xgb_h.pkl", 'rb'))
+loaded_model = pickle.load(open("heart_ba4522_example.pkl", 'rb'))
 
 # Setup SHAP
 explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
 
 # Create the main function for server
-def main_func(Generation,Tenure,Engagement, LearningDevelopment,WorkEnvironment, RewardsBenefits, EmployeeWellBeing):
-    new_row = pd.DataFrame.from_dict({'Generation': Generation,
-                                      'Tenure':Tenure,
-                                      'Engagement':Engagement,
-                                      'LearningDevelopment':LearningDevelopment,
-                                      'WorkEnvironment':WorkEnvironment,
-                                      'RewardsBenefits':RewardsBenefits,
-                                      'EmployeeWellBeing':EmployeeWellBeing}, orient = 'index').transpose()
+def main_func(age, sex, cp, trestbps, chol, fbs, restecg, thalach,
+       exang, oldpeak, slope, ca, thal):
+    new_row = pd.DataFrame.from_dict({'age': age,
+                                      'sex':sex,
+                                      'cp':cp,
+                                      'trestbps':trestbps,
+                                      'chol':chol,
+                                      'fbs':fbs,
+                                      'restecg':restecg,
+                                      'thalach':thalach,
+                                      'exang':exang,
+                                      'oldpeak':oldpeak,
+                                      'slope':slope,
+                                      'ca':ca,
+                                      'thal':thal
+                                     }, orient = 'index').transpose()
     
     prob = loaded_model.predict_proba(new_row)
     
@@ -55,13 +63,20 @@ with gr.Blocks(title=title) as demo:
     gr.Markdown("""---""")
     with gr.Row():        
         with gr.Column():
-            Generation = gr.Slider(label="Generation", minimum=1, maximum=6, value=4, step=1)
-            Tenure = gr.Slider(label="Tenure Level", minimum=1, maximum=8, value=4, step=1)
-            Engagement = gr.Slider(label="Engagement Score", minimum=1, maximum=5, value=4, step=.1)
-            LearningDevelopment = gr.Slider(label="LearningDevelopment Score", minimum=1, maximum=5, value=4, step=.1)
-            WorkEnvironment = gr.Slider(label="WorkEnvironment Score", minimum=1, maximum=5, value=4, step=.1)
-            RewardsBenefits = gr.Slider(label="RewardsBenefits Score", minimum=1, maximum=5, value=4, step=.1)
-            EmployeeWellBeing = gr.Slider(label="EmployeeWellBeing Score", minimum=1, maximum=5, value=4, step=.1) 
+            age = gr.Slider(label="age", minimum=1, maximum=6, value=4, step=1)
+            sex = gr.Slider(label="sex", minimum=1, maximum=8, value=4, step=1)
+            cp = gr.Slider(label="cp Score", minimum=1, maximum=5, value=4, step=.1)
+            trestbps = gr.Slider(label="trestbps Score", minimum=1, maximum=5, value=4, step=.1)
+            chol = gr.Slider(label="chol Score", minimum=1, maximum=5, value=4, step=.1)
+            fbs = gr.Slider(label="fbs Score", minimum=1, maximum=5, value=4, step=.1)
+            restecg = gr.Slider(label="restecg Score", minimum=1, maximum=5, value=4, step=.1)
+            thalach = gr.Slider(label="thalach Score", minimum=1, maximum=5, value=4, step=.1) 
+            exang = gr.Slider(label="exang Score", minimum=1, maximum=5, value=4, step=.1)
+            oldpeak = gr.Slider(label="oldpeak Score", minimum=1, maximum=5, value=4, step=.1)
+            slope = gr.Slider(label="slope Score", minimum=1, maximum=5, value=4, step=.1)
+            ca = gr.Slider(label="ca Score", minimum=1, maximum=5, value=4, step=.1)
+            thal = gr.Slider(label="thal Score", minimum=1, maximum=5, value=4, step=.1) 
+            
             submit_btn = gr.Button("Analyze")
         with gr.Column(visible=True,scale=1, min_width=600) as output_col:
             label = gr.Label(label = "Predicted Label")
@@ -69,13 +84,13 @@ with gr.Blocks(title=title) as demo:
         
             submit_btn.click(
                 main_func,
-                [Generation, Tenure,Engagement,LearningDevelopment,WorkEnvironment,RewardsBenefits,EmployeeWellBeing],
+                [age, sex,cp,trestbps,chol,fbs,restecg,thalach,exang,oldpeak,slope,ca,thal],
                 [label,local_plot], api_name="Employee_Turnover"
             )
     
     gr.Markdown("### Click on any of the examples below to see how it works:")
-    gr.Examples([[3,4,4,4,4,5,5], [4,5,4,5,4,4,5]], 
-                [Generation,Tenure,Engagement,LearningDevelopment,WorkEnvironment,RewardsBenefits,EmployeeWellBeing], 
+    gr.Examples([[33,0,1,100,230,1,1,150,0,.9,2,1,6], [39,1,0,170,200,1,1,150,0,1.4,2,1,6]], 
+                [age,sex,cp,trestbps,chol,fbs,restecg,thalach,exang,oldpeak,slope,ca,thal], 
                 [label,local_plot], main_func, cache_examples=True)
 
 demo.launch()