Olivier-52 commited on
Commit ·
af97d0a
1
Parent(s): ddc329c
Update app.py
Browse filesFix model loading issue
app.py
CHANGED
|
@@ -30,13 +30,14 @@ tags_metadata = [
|
|
| 30 |
]
|
| 31 |
|
| 32 |
load_dotenv()
|
| 33 |
-
|
| 34 |
AWS_ACCESS_KEY_ID = os.environ["AWS_ACCESS_KEY_ID"]
|
| 35 |
AWS_SECRET_ACCESS_KEY = os.environ["AWS_SECRET_ACCESS_KEY"]
|
|
|
|
| 36 |
MODEL_URI = "models:/climate-fake-news-detector-model-XGBoost-v1@production"
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
model = mlflow.sklearn.load_model(
|
| 40 |
|
| 41 |
app = FastAPI(
|
| 42 |
title="API for Climate Fake News Detector",
|
|
|
|
| 30 |
]
|
| 31 |
|
| 32 |
load_dotenv()
|
| 33 |
+
MLFLOW_TRACKING_APP_URI = os.environ["MLFLOW_TRACKING_APP_URI"]
|
| 34 |
AWS_ACCESS_KEY_ID = os.environ["AWS_ACCESS_KEY_ID"]
|
| 35 |
AWS_SECRET_ACCESS_KEY = os.environ["AWS_SECRET_ACCESS_KEY"]
|
| 36 |
+
|
| 37 |
MODEL_URI = "models:/climate-fake-news-detector-model-XGBoost-v1@production"
|
| 38 |
+
|
| 39 |
+
mlflow.set_tracking_uri(MLFLOW_TRACKING_APP_URI)
|
| 40 |
+
model = mlflow.sklearn.load_model(MODEL_URI)
|
| 41 |
|
| 42 |
app = FastAPI(
|
| 43 |
title="API for Climate Fake News Detector",
|