diff --git a/machine-learning/app/models/base.py b/machine-learning/app/models/base.py
index fd1be17e5b..b3eb6a48d4 100644
--- a/machine-learning/app/models/base.py
+++ b/machine-learning/app/models/base.py
@@ -5,12 +5,12 @@ from abc import ABC, abstractmethod
 from pathlib import Path
 from shutil import rmtree
 from typing import Any
-import onnx
-from onnx.tools.update_model_dims import update_inputs_outputs_dims
-from onnx.shape_inference import infer_shapes
 
+import onnx
 import onnxruntime as ort
 from huggingface_hub import snapshot_download
+from onnx.shape_inference import infer_shapes
+from onnx.tools.update_model_dims import update_inputs_outputs_dims
 from typing_extensions import Buffer
 
 import ann.ann