Fixing pseudo rgb

src/model/yolo_wrapper.py

@@ -257,17 +257,11 @@ class YOLOWrapper:
             if source_path.is_file():
                 try:
                     img_obj = Image(source_path)
-                    pil_img = img_obj.pil_image
-                    if len(pil_img.getbands()) == 1:
-                        rgb_img = img_obj.convert_grayscale_to_rgb_preserve_range()
-                    else:
-                        rgb_img = pil_img.convert("RGB")
-
                     suffix = source_path.suffix or ".png"
                     tmp = tempfile.NamedTemporaryFile(suffix=suffix, delete=False)
                     tmp_path = tmp.name
                     tmp.close()
-                    rgb_img.save(tmp_path)
+                    img_obj.save(tmp_path)
                     cleanup_path = tmp_path
                     logger.info(
                         f"Converted image {source_path} to RGB for inference at {tmp_path}"