Spaces:
Running
Running
Déploiement automatique depuis GitLab CI
Browse files
config.py
CHANGED
|
@@ -1,13 +1,24 @@
|
|
| 1 |
-
#
|
| 2 |
-
|
|
|
|
| 3 |
|
| 4 |
# Colonnes attendues par le modèle (features brutes)
|
| 5 |
FEATURE_COLUMNS = ["Area", "Item", "Year", "average_rain_fall_mm_per_year", "pesticides_tonnes", "avg_temp"]
|
| 6 |
TARGET_COLUMN = "hg/ha_yield"
|
| 7 |
|
|
|
|
|
|
|
|
|
|
| 8 |
# Chemin relatif vers le modèle MLflow (pipeline sklearn complet avec préprocesseur)
|
| 9 |
MODEL_REL_PATH = "5_Notebooks/mlruns/2/models/m-7d0b2bca11e140be907efb132e749af8/artifacts"
|
| 10 |
|
| 11 |
-
#
|
| 12 |
-
|
|
|
|
| 13 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# ========================================================================
|
| 2 |
+
# Paramètres pour le modèle
|
| 3 |
+
# ========================================================================
|
| 4 |
|
| 5 |
# Colonnes attendues par le modèle (features brutes)
|
| 6 |
FEATURE_COLUMNS = ["Area", "Item", "Year", "average_rain_fall_mm_per_year", "pesticides_tonnes", "avg_temp"]
|
| 7 |
TARGET_COLUMN = "hg/ha_yield"
|
| 8 |
|
| 9 |
+
# Définition de la séparation de la Timeseries en Train et Test sets pour les différents modèles de prédiction
|
| 10 |
+
YEAR_THRESHOLD = 2009
|
| 11 |
+
|
| 12 |
# Chemin relatif vers le modèle MLflow (pipeline sklearn complet avec préprocesseur)
|
| 13 |
MODEL_REL_PATH = "5_Notebooks/mlruns/2/models/m-7d0b2bca11e140be907efb132e749af8/artifacts"
|
| 14 |
|
| 15 |
+
# ========================================================================
|
| 16 |
+
# Paramètres pour les services (Frontend et Backend)
|
| 17 |
+
# ========================================================================
|
| 18 |
|
| 19 |
+
# URLs des services pour le frontend (Streamlit) et le backend (API - Streamlit)
|
| 20 |
+
DEFAULT_FRONTEND_URL = "https://cedm-oc-mlops-projet-4-dashboard.hf.space"
|
| 21 |
+
DEFAULT_API_URL = "https://cedm-oc-mlops-projet-4-api.hf.space"
|
| 22 |
+
|
| 23 |
+
# Chemin relatif vers le fichier d'entraînement nettoyé
|
| 24 |
+
DATA_REL_PATH = "2_Data_transformed/crop_yield_2_train_set_simplified.csv"
|
main.py
CHANGED
|
@@ -19,7 +19,7 @@ from dotenv import load_dotenv
|
|
| 19 |
from fastapi import FastAPI, HTTPException
|
| 20 |
from pydantic import BaseModel, Field
|
| 21 |
|
| 22 |
-
from config import FEATURE_COLUMNS, TARGET_COLUMN, MODEL_REL_PATH, DATA_REL_PATH
|
| 23 |
|
| 24 |
# Charger les variables d'environnement depuis .env
|
| 25 |
load_dotenv()
|
|
@@ -40,8 +40,8 @@ logfire.configure(
|
|
| 40 |
|
| 41 |
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
| 42 |
|
| 43 |
-
# URL du frontend à
|
| 44 |
-
FRONTEND_URL = os.environ.get("FRONTEND_URL",
|
| 45 |
|
| 46 |
# Chemin vers le modèle MLflow (pipeline sklearn complet avec préprocesseur)
|
| 47 |
MODEL_PATH = os.path.join(BASE_DIR, MODEL_REL_PATH)
|
|
|
|
| 19 |
from fastapi import FastAPI, HTTPException
|
| 20 |
from pydantic import BaseModel, Field
|
| 21 |
|
| 22 |
+
from config import FEATURE_COLUMNS, TARGET_COLUMN, MODEL_REL_PATH, DATA_REL_PATH, DEFAULT_FRONTEND_URL
|
| 23 |
|
| 24 |
# Charger les variables d'environnement depuis .env
|
| 25 |
load_dotenv()
|
|
|
|
| 40 |
|
| 41 |
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
| 42 |
|
| 43 |
+
# URL du frontend à réveiller toutes les 12h (variable d'env prioritaire, sinon valeur de config.py)
|
| 44 |
+
FRONTEND_URL = os.environ.get("FRONTEND_URL", DEFAULT_FRONTEND_URL)
|
| 45 |
|
| 46 |
# Chemin vers le modèle MLflow (pipeline sklearn complet avec préprocesseur)
|
| 47 |
MODEL_PATH = os.path.join(BASE_DIR, MODEL_REL_PATH)
|