pradelf commited on
Commit
c8447c0
·
verified ·
1 Parent(s): 45ba6f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -11
app.py CHANGED
@@ -70,18 +70,8 @@ tags_metadata = [
70
  },
71
  ]
72
 
73
- @app.on_event("startup")
74
- def load_model_on_startup():
75
- global model
76
- try:
77
- logger.info(f"Chargement du modèle depuis {MLFLOW_TRACKING_URI}")
78
- logger.info(f"MLFLOW_TRACKING_URI={os.getenv('MLFLOW_TRACKING_URI')}")
79
- model = mlflow.pyfunc.load_model(MODEL_URI)
80
- logger.info("Modèle chargé avec succès")
81
- except Exception:
82
- logger.exception("Impossible de charger le modèle au démarrage")
83
- model = None
84
 
 
85
 
86
  app = FastAPI(
87
  title="Getaround API pour le prix journalier de location d'une voiture.",
@@ -94,6 +84,17 @@ app = FastAPI(
94
  openapi_tags=tags_metadata,
95
  )
96
 
 
 
 
 
 
 
 
 
 
 
 
97
 
98
  class RentalFeatures(BaseModel):
99
  model_key: str
 
70
  },
71
  ]
72
 
 
 
 
 
 
 
 
 
 
 
 
73
 
74
+ model= None
75
 
76
  app = FastAPI(
77
  title="Getaround API pour le prix journalier de location d'une voiture.",
 
84
  openapi_tags=tags_metadata,
85
  )
86
 
87
+ @app.on_event("startup")
88
+ def load_model_on_startup():
89
+ global model
90
+ try:
91
+ logger.info(f"Chargement du modèle depuis {MLFLOW_TRACKING_URI}")
92
+ logger.info(f"MLFLOW_TRACKING_URI={os.getenv('MLFLOW_TRACKING_URI')}")
93
+ model = mlflow.pyfunc.load_model(MODEL_URI)
94
+ logger.info("Modèle chargé avec succès")
95
+ except Exception:
96
+ logger.exception("Impossible de charger le modèle au démarrage")
97
+ model = None
98
 
99
  class RentalFeatures(BaseModel):
100
  model_key: str