diff --git a/nvidia-clara-agx/colonoscopy_segmentation/colonoscopy_segmentation.py b/nvidia-clara-agx/colonoscopy_segmentation/colonoscopy_segmentation.py
index 4705f88..21a2b13 100644
--- a/nvidia-clara-agx/colonoscopy_segmentation/colonoscopy_segmentation.py
+++ b/nvidia-clara-agx/colonoscopy_segmentation/colonoscopy_segmentation.py
@@ -54,8 +54,9 @@ def __init__(self, data, source="replayer"):
         self.sample_data_path = data
 
         self.model_path_map = {
-            "ultrasound_seg": os.path.join(self.sample_data_path, "ColonSegNet-07112023-2359.onnx"),
+            #"ultrasound_seg": os.path.join(self.sample_data_path, "ColonSegNet-07112023-2359.onnx"),
             #"ultrasound_seg": os.path.join(self.sample_data_path, "colon.onnx"),
+            "ultrasound_seg": os.path.join(self.sample_data_path, "ColonSegNet_brightcontr.onnx"),
         }
 
     def compose(self):