Update inference.py

inference.py

@@ -1,8 +1,9 @@
 import joblib
 import pandas as pd
 
-# Load the model
+# Load the model and columns
 model = joblib.load("time_slot_model.joblib")
+columns = joblib.load("columns.joblib")
 
 def preprocess_time(time_str):
     hour = int(time_str.split(':')[0])
@@ -18,15 +19,18 @@ def categorize_time(hour):
     elif 12 <= hour < 14:
         return 'early_noon'
     elif 14 <= hour < 16:
-        return 'early_noon'
+        return 'noon'
     elif 16 <= hour < 17:
         return 'dusk'
     else:
         return 'other'
 
-def predict(user_input):
-    # Preprocess and predict
+def predict(user_name, user_input):
+    # Process the user input
     user_input = pd.get_dummies(pd.Series([user_input]), prefix='Pref')
-    user_input = user_input.reindex(columns=X.columns, fill_value=0)
+    user_input = user_input.reindex(columns=columns, fill_value=0)
+
+    # Predict the preferred time slot
     prediction = model.predict(user_input)
     return prediction[0]
+