download model with joblib instead of pickle

main.py

@@ -7,6 +7,7 @@ from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 import pickle
+import joblib
 from extraction_features import extract_features 
 
 app = FastAPI()
@@ -20,8 +21,9 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-model_file = open('logistic_regression_model.pkl', 'rb')
-model = pickle.load(model_file, encoding='bytes')
+# model_file = open('logistic_regression_model.pkl', 'rb')
+# model = pickle.load(model_file, encoding='bytes')
+model = joblib.load('logistic_regression_model.pkl')
 
 
 class Msg(BaseModel):