upate

src/model/yolo_wrapper.py

@@ -9,7 +9,7 @@ from typing import Optional, List, Dict, Callable, Any
 import torch
 import tempfile
 import os
-from src.utils.image import Image, convert_grayscale_to_rgb_preserve_range
+from src.utils.image import Image
 from src.utils.logger import get_logger
 
 
@@ -238,7 +238,7 @@ class YOLOWrapper:
                     img_obj = Image(source_path)
                     pil_img = img_obj.pil_image
                     if len(pil_img.getbands()) == 1:
-                        rgb_img = convert_grayscale_to_rgb_preserve_range(pil_img)
+                        rgb_img = img_obj.convert_grayscale_to_rgb_preserve_range()
                     else:
                         rgb_img = pil_img.convert("RGB")