Olivier-52 commited on
Commit ·
b272ac7
1
Parent(s): af97d0a
Update app.py to resolve credentials
Browse files
app.py
CHANGED
|
@@ -5,7 +5,6 @@ from pydantic import BaseModel
|
|
| 5 |
from fastapi import FastAPI, File, UploadFile
|
| 6 |
import mlflow
|
| 7 |
from xgboost import XGBClassifier
|
| 8 |
-
import os
|
| 9 |
from dotenv import load_dotenv
|
| 10 |
|
| 11 |
description = """
|
|
@@ -30,14 +29,13 @@ tags_metadata = [
|
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
| 38 |
|
| 39 |
-
mlflow.
|
| 40 |
-
model = mlflow.sklearn.load_model(MODEL_URI)
|
| 41 |
|
| 42 |
app = FastAPI(
|
| 43 |
title="API for Climate Fake News Detector",
|
|
@@ -86,4 +84,4 @@ def predict(features: PredictionFeatures):
|
|
| 86 |
return {"prediction": float(prediction)}
|
| 87 |
|
| 88 |
if __name__ == "__main__":
|
| 89 |
-
uvicorn.run(app, host="localhost", port=
|
|
|
|
| 5 |
from fastapi import FastAPI, File, UploadFile
|
| 6 |
import mlflow
|
| 7 |
from xgboost import XGBClassifier
|
|
|
|
| 8 |
from dotenv import load_dotenv
|
| 9 |
|
| 10 |
description = """
|
|
|
|
| 29 |
]
|
| 30 |
|
| 31 |
load_dotenv()
|
|
|
|
|
|
|
|
|
|
| 32 |
|
| 33 |
+
os.environ["AWS_ACCESS_KEY_ID"] = os.getenv("AWS_ACCESS_KEY_ID")
|
| 34 |
+
os.environ["AWS_SECRET_ACCESS_KEY"] = os.getenv("AWS_SECRET_ACCESS_KEY")
|
| 35 |
+
|
| 36 |
+
mlflow.set_tracking_uri("https://olivier-52-ml-flow.hf.space")
|
| 37 |
|
| 38 |
+
model = mlflow.sklearn.load_model("models:/climate-fake-news-detector-model-XGBoost-v1@production")
|
|
|
|
| 39 |
|
| 40 |
app = FastAPI(
|
| 41 |
title="API for Climate Fake News Detector",
|
|
|
|
| 84 |
return {"prediction": float(prediction)}
|
| 85 |
|
| 86 |
if __name__ == "__main__":
|
| 87 |
+
uvicorn.run(app, host="localhost", port=8001)
|