diff --git a/image_prediction/predictor.py b/image_prediction/predictor.py
index d994baf..4450e1a 100644
--- a/image_prediction/predictor.py
+++ b/image_prediction/predictor.py
@@ -27,9 +27,7 @@ class Predictor:
         """
         try:
             if model_handle is None:
-                reader = MlflowModelReader(
-                    run_id=CONFIG.service.run_id, mlruns_dir=MLRUNS_DIR
-                )
+                reader = MlflowModelReader(run_id=CONFIG.service.run_id, mlruns_dir=MLRUNS_DIR)
                 self.model_handle = reader.get_model_handle(BASE_WEIGHTS)
             else:
                 self.model_handle = model_handle