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

Update app.py

Browse files

Fix model loading issue

Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -30,13 +30,14 @@ tags_metadata = [
30
  ]
31
 
32
  load_dotenv()
33
- MLFLOW_TRACKING_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
  MODEL_URI = "models:/climate-fake-news-detector-model-XGBoost-v1@production"
37
- mlflow.set_tracking_uri(MLFLOW_TRACKING_URI)
38
- model_uri = MODEL_URI
39
- model = mlflow.sklearn.load_model(model_uri)
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",