Update app.py

app.py

@@ -1,4 +1,3 @@
-import pickle
 import pandas as pd
 import shap
 import gradio as gr
@@ -21,6 +20,7 @@ def safe_convert(value, default, min_val, max_val):
 # Create the main function for the model
 def main_func(Department, ChainScale, SupportiveGM, Merit, LearningDevelopment, WorkEnvironment, Engagement, WellBeing):
     
+    
     ChainScale_mapping = {
         'Luxury': 1,
         'Upper Midscale': 2,
@@ -36,12 +36,8 @@ def main_func(Department, ChainScale, SupportiveGM, Merit, LearningDevelopment,
         "Front Office Operations": 4,
         "Guest Activities": 5,
     }
-    
-    # Convert inputs
-    Department = department_mapping.get(Department, 1)  # Default to "Guest Services"
-    ChainScale = ChainScale_mapping.get(ChainScale, 3)  # Default to "Upper Upscale"
 
-    # Ensure numeric input validity
+    
     LearningDevelopment = safe_convert(LearningDevelopment, 3.0, 1, 5)
     SupportiveGM = safe_convert(SupportiveGM, 3.0, 1, 5)
     Merit = safe_convert(Merit, 3.0, 1, 5)
@@ -49,10 +45,8 @@ def main_func(Department, ChainScale, SupportiveGM, Merit, LearningDevelopment,
     Engagement = safe_convert(Engagement, 3.0, 1, 5)
     WellBeing = safe_convert(WellBeing, 3.0, 1, 5)
 
-    # Create DataFrame for prediction
+    
     new_row = pd.DataFrame({
-        'Department': [Department],
-        'ChainScale': [ChainScale],
         'SupportiveGM': [SupportiveGM],
         'Merit': [Merit],
         'LearningDevelopment': [LearningDevelopment],
@@ -162,15 +156,13 @@ with gr.Blocks(title=title) as demo:
     
     gr.Examples(
         [
-            ["Guest Services", "Upper Upscale", 4.1, 3.7, 3.9, 4.2, 4.4, 4.3],  
+            ["Guest Service", "Upper Upscale", 4.1, 3.7, 3.9, 4.2, 4.4, 4.3],  
             ["Food and Beverage", "Upper Upscale", 3.9, 3.7, 4.1, 4.3, 4.5, 4.4],  
             ["Housekeeping", "Upper Upscale", 4.3, 4.0, 4.3, 4.4, 4.5, 4.4],  
-            ["Guest Services", "Upper Upscale", 5.0, 4.0, 4.3, 4.4, 5.0, 5.0]  
         ],
         [Department, ChainScale, SupportiveGM, Merit, LearningDevelopment, WorkEnvironment, Engagement, WellBeing],
         [label, local_plot],
         main_func,
         cache_examples=True
     )
-    
 demo.launch()