From b430a4409626d04f146d4e0ebd995004a3c24723 Mon Sep 17 00:00:00 2001
From: Juhi Manish Jain <97155012+juhimanishjain@users.noreply.github.com>
Date: Mon, 7 Apr 2025 12:08:12 +0530
Subject: [PATCH] Update model.py

---
 src/onnx_models/age_classify_v001/model.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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