modified models.py

models.py

@@ -30,8 +30,8 @@ def save_model(model, filename):
 
 
 def load_model(filename):
-    loaded_model = joblib.load('log_reg_model.pkl')
-    # with open(filename, 'rb') as model_file:
-        # loaded_model = pickle.load(model_file)
+    # loaded_model = joblib.load('log_reg_model.pkl')
+    with open(filename, 'rb') as model_file:
+        loaded_model = pickle.load(model_file)
     print(f"Model loaded from {filename}")
     return loaded_model