diff --git a/image_prediction/redai_adapter/efficient_net_wrapper.py b/image_prediction/redai_adapter/efficient_net_wrapper.py
index f43a1b7..d35b20c 100644
--- a/image_prediction/redai_adapter/efficient_net_wrapper.py
+++ b/image_prediction/redai_adapter/efficient_net_wrapper.py
@@ -1,11 +1,7 @@
-import os
+import tensorflow as tf
 
 from image_prediction.redai_adapter.model_wrapper import ModelWrapper
 
-os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
-
-import tensorflow as tf
-
 
 class EfficientNetWrapper(ModelWrapper):
 
diff --git a/image_prediction/redai_adapter/mlflow.py b/image_prediction/redai_adapter/mlflow.py
index 55ceab3..7930ae3 100644
--- a/image_prediction/redai_adapter/mlflow.py
+++ b/image_prediction/redai_adapter/mlflow.py
@@ -31,7 +31,6 @@ class MlflowModelReader:
 
     @lru_cache(maxsize=None)
     def __get_run(self, run_id):
-
         return mlflow.get_run(run_id)
 
     def __get_classes(self, run_id, prefix="tt"):
diff --git a/image_prediction/redai_adapter/model_wrapper.py b/image_prediction/redai_adapter/model_wrapper.py
index 7a50f53..c31f4fa 100644
--- a/image_prediction/redai_adapter/model_wrapper.py
+++ b/image_prediction/redai_adapter/model_wrapper.py
@@ -1,10 +1,6 @@
 import abc
-import os
-
-os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
 
 import numpy as np
-
 import tensorflow as tf