Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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)
|