Yair - Added error handling for XGB model

model_manager.py

@@ -6,7 +6,9 @@ from config import CATBOOST_MODEL_PATH, XGB_MODEL_PATH, RF_MODEL_PATH
 def save_models(models):
     """ Save trained models """
     models["CatBoost"].save_model(CATBOOST_MODEL_PATH)
-    models["XGBoost"].save_model(XGB_MODEL_PATH)
+    if models["XGBoost"] is not None:
+        # Save XGBoost model in binary format to reduce memory usage
+        models["XGBoost"].get_booster().save_model(XGB_MODEL_PATH)
     joblib.dump(models["RandomForest"], RF_MODEL_PATH)
     print("✅ Models saved successfully!")
 
@@ -15,9 +17,9 @@ def load_models():
     catboost = CatBoostClassifier()
     catboost.load_model(CATBOOST_MODEL_PATH)
 
-    xgb = XGBClassifier()
+    xgb = XGBClassifier()  # Load XGBoost model in binary format
     xgb.load_model(XGB_MODEL_PATH)
 
     rf = joblib.load(RF_MODEL_PATH)
 
-    return {"CatBoost": catboost, "XGBoost": xgb, "RandomForest": rf}
+    return {"CatBoost": catboost, "XGBoost": xgb, "RandomForest": rf}