diff --git a/src/deepness/processing/models/model_base.py b/src/deepness/processing/models/model_base.py
index 88c8cc5..f861267 100644
--- a/src/deepness/processing/models/model_base.py
+++ b/src/deepness/processing/models/model_base.py
@@ -3,7 +3,6 @@
 import json
 from typing import List, Optional
 
-import cv2
 import numpy as np
 
 import deepness.processing.models.preprocessing_utils as preprocessing_utils
@@ -84,7 +83,7 @@ def get_model_batch_size(self) -> Optional[int]:
             Batch size or None if not found (dynamic batch size)
         """
         bs = self.input_shape[0]
-                
+
         if isinstance(bs, str):
             return None
         else:
@@ -183,7 +182,7 @@ def get_metadata_standarization_parameters(self) -> Optional[StandardizationPara
         meta = self.sess.get_modelmeta()
         name_mean = 'standardization_mean'
         name_std = 'standardization_std'
-        
+
         param = StandardizationParameters(channels_number=self.get_input_shape()[-3])
 
         if name_mean in meta.custom_metadata_map and name_std in meta.custom_metadata_map: