ericjedha commited on
Commit
c7026b9
·
verified ·
1 Parent(s): 511986a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -5
app.py CHANGED
@@ -10,15 +10,32 @@ from contextlib import asynccontextmanager
10
 
11
  # --- Configuration ---
12
 
 
 
 
 
13
  os.environ["APP_URI"] = "https://ericjedha-getaround.hf.space"
 
 
 
 
 
14
  mlflow.set_tracking_uri(os.environ["APP_URI"])
 
 
 
 
 
15
  logged_model = 'runs:/f7019001dcb440f9b4a739550c670dba/best_estimator'
16
- loaded_model = mlflow.pyfunc.load_model(logged_model)
 
 
 
 
 
 
 
17
 
18
- if(loaded_model):
19
- print("OK")
20
- else:
21
- print("NOK")
22
 
23
  # Configuration des logs
24
  logging.basicConfig(level=logging.INFO)
 
10
 
11
  # --- Configuration ---
12
 
13
+ import os
14
+ import mlflow
15
+
16
+ # Définir la variable d'environnement
17
  os.environ["APP_URI"] = "https://ericjedha-getaround.hf.space"
18
+
19
+ # Vérifier la valeur de la variable d'environnement
20
+ print("APP_URI:", os.environ["APP_URI"])
21
+
22
+ # Définir l'URI de suivi MLflow
23
  mlflow.set_tracking_uri(os.environ["APP_URI"])
24
+
25
+ # Vérifier l'URI de suivi MLflow
26
+ print("MLflow Tracking URI:", mlflow.get_tracking_uri())
27
+
28
+ # Charger le modèle
29
  logged_model = 'runs:/f7019001dcb440f9b4a739550c670dba/best_estimator'
30
+ try:
31
+ loaded_model = mlflow.pyfunc.load_model(logged_model)
32
+ if loaded_model:
33
+ print("OK")
34
+ else:
35
+ print("NOK")
36
+ except Exception as e:
37
+ print("Erreur lors du chargement du modèle:", e)
38
 
 
 
 
 
39
 
40
  # Configuration des logs
41
  logging.basicConfig(level=logging.INFO)