added joblib library

models.py

@@ -2,6 +2,7 @@ from sentence_transformers import SentenceTransformer
 from sklearn.linear_model import LogisticRegression
 import pickle
 from sklearn.model_selection import train_test_split
+import joblib
 
 import pandas as pd
 
@@ -29,7 +30,8 @@ def save_model(model, filename):
 
 
 def load_model(filename):
-    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