Spaces:

GhylB
/

sepsis_prediction_app

Runtime error

GhylB commited on Jun 6, 2023

Commit

6f81365

1 Parent(s): 6c72ddb

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -17,7 +17,7 @@ app = FastAPI(title="API")
 """We load a machine learning model and a scaler that help us make predictions based on data."""
 model = joblib.load('gbc.pkl',mmap_mode='r')
-scaler = joblib.load('scaler.pkl')
 """We define a function that will make predictions using our model and scaler."""
 def predict(df, endpoint='simple'):

 """We load a machine learning model and a scaler that help us make predictions based on data."""
 model = joblib.load('gbc.pkl',mmap_mode='r')
+scaler = joblib.load('scaler.pkl',mmap_mode='r')
 """We define a function that will make predictions using our model and scaler."""
 def predict(df, endpoint='simple'):