diff --git a/src/onnx_models/age_classify_v001/model.py b/src/onnx_models/age_classify_v001/model.py
index d5cf353..70ee339 100644
--- a/src/onnx_models/age_classify_v001/model.py
+++ b/src/onnx_models/age_classify_v001/model.py
@@ -1,5 +1,6 @@
 import onnxruntime as ort, os
-import numpy as np, cv2 as cv
+import numpy as np
+import cv2 as cv
 from PIL import Image
 from src.utils.preprocess import enhance_image
 
@@ -68,4 +69,4 @@ def predict(image_input, onnx_path):
     predicted_class = predicted_classes[0]
     predicted_label = id2label.get(predicted_class, "Unknown")
     confidence = probabilities[0][predicted_class]
-    return predicted_label, confidence
\ No newline at end of file
+    return predicted_label, confidence