Olivier-52 commited on
Commit
b272ac7
·
1 Parent(s): af97d0a

Update app.py to resolve credentials

Browse files
Files changed (1) hide show
  1. app.py +6 -8
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
- 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",
@@ -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=8000)
 
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)