Olivier-52 commited on
Commit
2741c49
·
1 Parent(s): 5e295a6

Update app.py

Browse files

add logs for model load

Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -43,6 +43,10 @@ os.environ["AWS_SECRET_ACCESS_KEY"] = os.getenv("AWS_SECRET_ACCESS_KEY")
43
  mlflow.set_tracking_uri(MLFLOW_TRACKING_APP_URI)
44
  model_uri = f"models:/{MODEL_NAME}@{STAGE}"
45
  model = mlflow.sklearn.load_model(model_uri)
 
 
 
 
46
 
47
  app = FastAPI(
48
  title="API for Fraud Detection Project",
 
43
  mlflow.set_tracking_uri(MLFLOW_TRACKING_APP_URI)
44
  model_uri = f"models:/{MODEL_NAME}@{STAGE}"
45
  model = mlflow.sklearn.load_model(model_uri)
46
+ if model is None:
47
+ raise ValueError("Model not found.")
48
+ else:
49
+ print("Model loaded.")
50
 
51
  app = FastAPI(
52
  title="API for Fraud Detection Project",