.

model_handler.py

@@ -99,13 +99,22 @@ class AIDetectionModelHandler:
         self.xgboost_model = None
         self.model_loaded = False
         
-        # Default model paths
+             # Default model paths
         if model_path is None:
-            backend_dir = Path(__file__).parent
-            model_path = str(backend_dir.parent / "model" / "model")
+            # Prefer explicit env var
+            env_model_path = os.getenv("MODEL_PATH")
+            if env_model_path and os.path.exists(env_model_path):
+                model_path = env_model_path
+            elif os.path.exists("/app/model"):
+                model_path = "/app/model"
+            else:
+                # Fallback to legacy relative path
+                backend_dir = Path(__file__).parent
+                model_path = str(backend_dir.parent / "model" / "model")
         
         self.model_path = model_path
-        self.xgboost_path = str(Path(model_path).parent / "xgboost_model.json")
+        # XGBoost file is expected inside the same folder as the other model artifacts
+        self.xgboost_path = str(Path(model_path) / "xgboost_model.json")
         
         # Load the models
         self._load_models()