Merge pull request #39 from marintosti12/feature/loader-model-env

.github/workflows/ci.yaml

@@ -18,6 +18,7 @@ jobs:
           POSTGRES_USER: ci
           POSTGRES_PASSWORD: ci
           POSTGRES_DB: futurisys_ci
+
         ports:
           - 5432:5432
         options: >-
@@ -29,6 +30,8 @@ jobs:
     env:
       DATABASE_URL: postgresql+asyncpg://ci:ci@localhost:5432/futurisys_ci
       HF_TOKEN: ${{ secrets.HF_TOKEN }}
+      APP_ENV: test
+
 
     steps:
       - name: Checkout

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"):
+        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)