CedM commited on
Commit
a4c39e7
·
verified ·
1 Parent(s): 3efdd60

Déploiement automatique depuis GitLab CI

Browse files
Files changed (2) hide show
  1. config.py +15 -4
  2. main.py +3 -3
config.py CHANGED
@@ -1,13 +1,24 @@
1
- # Définition de la séparation de la Timeseries en Train et Test sets pour les différents modèles de prédiction
2
- YEAR_THRESHOLD = 2009
 
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
- # Chemin relatif vers le fichier d'entraînement nettoyé
12
- DATA_REL_PATH = "2_Data_transformed/crop_yield_2_train_set_simplified.csv"
 
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 à reveiller toutes les 12h (définie via variable d'environnement)
44
- FRONTEND_URL = os.environ.get("FRONTEND_URL", "https://huggingface.co/spaces/CedM/oc_mlops_projet_4_dashboard")
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)