Update main.py

main.py

@@ -63,7 +63,7 @@ def predict(
     pred_proba = xgb_model.predict_proba(input_df)
 
     # Output
-    predicted_status = "Unknown" if pred[0] == -1 else encoders['status.name'].inverse_transform([pred])[0]
+    predicted_status = "Unknown" if pred[0] == -1 else encoders['status_name'].inverse_transform([pred])[0]
     probability = pred_proba[0][pred[0]] * 100 if pred[0] != -1 else "Unknown"
 
     if predicted_status == "Returned to Client":