feat (model loader) : update loader for model in local

README.md

@@ -58,7 +58,7 @@ HF_TOKEN= Token Hugging Face
 ### 4. Base de données (PostgreSQL)
 
 ~~~bash
-docker compose up -d
+sudo docker compose up -d
 ~~~
 
 

src/model_loader.py

@@ -1,18 +1,35 @@
 import os
 from functools import lru_cache
-from typing import Any
+from pathlib import Path
+from typing import Any, Literal
 from huggingface_hub import hf_hub_download
 import joblib
 
 HF_REPO_ID  = os.getenv("HF_REPO_ID",  "Marintosti/attrition")
 HF_TOKEN    = os.getenv("HF_TOKEN")      
 
+ENV: Literal["dev", "test", "prod"] = os.getenv("APP_ENV", "dev").lower()
+ARTIFACTS_DIR = Path(os.getenv("ARTIFACTS_DIR", "artifacts"))
+
+def _load_local(name: str) -> Any:
+    path = ARTIFACTS_DIR / f"{name}.joblib"
+    if not path.exists():
+        raise FileNotFoundError(
+            f"Modèle local introuvable: {path}. "
+        )
+    return joblib.load(path)
+
 @lru_cache(maxsize=1)
 def load_model(name) -> Any:
-    local_path = hf_hub_download(
+
+    if ENV in ("dev", "test"):
+        return _load_local(name)
+     
+    hf_path = hf_hub_download(
         repo_id=HF_REPO_ID,
         filename=f"{name}.joblib",
         token=HF_TOKEN,
         local_files_only=False,
     )
-    return joblib.load(local_path)
+
+    return joblib.load(hf_path)