Merge pull request #64 from McGill-MMA-EnterpriseAnalytics/aladelca-p…

…atch-30 Update predict.py
McGill-MMA-EnterpriseAnalytics · Apr 26, 2024 · f23cb8c · f23cb8c
2 parents e83a1fa + f4b80cd
commit f23cb8c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/09_model_pipeline/predict/src/predict.py b/09_model_pipeline/predict/src/predict.py
@@ -3,8 +3,8 @@
 import pickle
 class Predict:
     def main_predict(data, bucket_name, path):
-        #model = load_models_from_s3(bucket_name,  path )
-        model = pickle.load(open("09_model_pipeline/predict/src/trained_model.pickle",'rb'))
+        model = load_models_from_s3(bucket_name,  path )
+        #model = pickle.load(open("09_model_pipeline/predict/src/trained_model.pickle",'rb'))
         probas = model.predict_proba(data)
         preds = np.where(probas[:,1] >= 0.4, 1, 0)
         data['prob_0'] = probas[:,0]