Update app.py

app.py

@@ -7,7 +7,7 @@ import numpy as np
 import matplotlib.pyplot as plt
 
 # load the model from disk
-loaded_model = pickle.load(open("h22_xgb.pkl", 'rb'))
+loaded_model = pickle.load(open("XGBclf_v0.2.sav", 'rb'))
 
 # Setup SHAP
 explainer = shap.Explainer(loaded_model) # PLEASE DO NOT CHANGE THIS.
@@ -34,7 +34,7 @@ def main_func(ValueDiversity,AdequateResources,Voice,GrowthAdvancement,Workload,
 # Create the UI
 title = "**Employee Turnover Predictor & Interpreter** 🪐"
 description1 = """
-This app takes six inputs about employees' satisfaction with different aspects of their work (such as work-life balance, ...) and predicts whether the employee intends to stay with the employer or leave. There are two outputs from the app: 1- the predicted probability of stay or leave, 2- Shapley's force-plot which visualizes the extent to which each factor impacts the stay/ leave prediction.   
+This app takes five inputs about employees' satisfaction with different aspects of their work (such as job satisfaction, ...) and predicts whether the employee intends to stay with the employer or leave. There are two outputs from the app: 1- the predicted probability of stay or leave, 2- Shapley's force-plot which visualizes the extent to which each factor impacts the stay/ leave prediction.   
 """
 
 description2 = """
@@ -50,12 +50,11 @@ with gr.Blocks(title=title) as demo:
     gr.Markdown("""---""")
     with gr.Row():        
         with gr.Column():
-            ValueDiversity = gr.Slider(label="ValueDiversity Score", minimum=1, maximum=5, value=4, step=.1)
-            AdequateResources = gr.Slider(label="AdequateResources Score", minimum=1, maximum=5, value=4, step=.1)
-            Voice = gr.Slider(label="Voice Score", minimum=1, maximum=5, value=4, step=.1)
-            GrowthAdvancement = gr.Slider(label="GrowthAdvancement Score", minimum=1, maximum=5, value=4, step=.1)
-            Workload = gr.Slider(label="Workload Score", minimum=1, maximum=5, value=4, step=.1)
-            WorkLifeBalance = gr.Slider(label="WorkLifeBalance Score", minimum=1, maximum=5, value=4, step=.1)
+            Management = gr.Slider(label="Management Score", minimum=1, maximum=5, value=4, step=.1)
+            EmployeeWellBeing = gr.Slider(label="Employee Well Being Score", minimum=1, maximum=5, value=4, step=.1)
+            EngagedAtWork = gr.Slider(label="Work Engagement Score", minimum=1, maximum=5, value=4, step=.1)
+            WorkEnvironment = gr.Slider(label="Work Environment Score", minimum=1, maximum=5, value=4, step=.1)
+            JobSatisfaction = gr.Slider(label="Job Satisfaction Score", minimum=1, maximum=5, value=4, step=.1)
             submit_btn = gr.Button("Analyze")
         with gr.Column(visible=True) as output_col:
             label = gr.Label(label = "Predicted Label")
@@ -69,7 +68,7 @@ with gr.Blocks(title=title) as demo:
     
     gr.Markdown("### Click on any of the examples below to see how it works:")
     gr.Examples([[4,4,4,4,5,5], [5,4,5,4,4,4]], 
-                [ValueDiversity,AdequateResources,Voice,GrowthAdvancement,Workload,WorkLifeBalance], 
+                [Management,EmployeeWellBeing,EngagedAtWork,WorkEnvironment,JobSatisfaction], 
                 [label,local_plot], main_func, cache_examples=True)
 
 demo.launch()