fix seg app

app.py

@@ -3371,8 +3371,8 @@ with demo:
                         combined_mask = noisy_mask * 0.4 + clean_mask
                         combined_mask = np.clip(combined_mask, 0, 255).astype(np.uint8)
                         if overlay_image:
-                            combined_mask[:, :, 0] = 0  # remove red channel
-                            combined_mask[:, :, 1] = 0  # remove green channel
+                            # add empty red and green channel
+                            combined_mask = np.stack([np.zeros_like(combined_mask), np.zeros_like(combined_mask), combined_mask], axis=-1)
                             _image = original_images[i_image].convert("RGB").resize((combined_mask.shape[1], combined_mask.shape[0]))
                             _image = np.array(_image)
                             combined_mask = 0.5 * combined_mask + 0.5 * _image