Spaces:
Sleeping
Sleeping
Olivier-52 commited on
Commit ·
2741c49
1
Parent(s): 5e295a6
Update app.py
Browse filesadd logs for model load
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",
|