Update app.py

app.py

@@ -95,6 +95,7 @@ def render_segmentation(segmentation_logits, dataset):
     colormap_array = np.array(colormap, dtype=np.uint8)
     segmentation_logits += 1
     segmentation_values = colormap_array[segmentation_logits]
+    segmentation_values = segmentation_values[:, :, ::-1]
     unique_labels = np.unique(segmentation_logits)
 
     colormap_array = colormap_array[unique_labels]
@@ -106,7 +107,7 @@ def render_segmentation(segmentation_logits, dataset):
     for idx, color in enumerate(colormap_array):
         color_box = np.zeros((20, 20, 3), dtype=np.uint8)
         color_box[:, :] = color
-        color_box = cv2.cvtColor(color_box, cv2.COLOR_RGB2BGR)
+        # color_box = cv2.cvtColor(color_box, cv2.COLOR_RGB2BGR)
         _, img_data = cv2.imencode(".jpg", color_box)
         img_base64 = base64.b64encode(img_data).decode("utf-8")
         img_data_uri = f"data:image/jpg;base64,{img_base64}"