Change the interface output to 01 mask

mlcd_seg.py

@@ -1020,7 +1020,8 @@ class MLCDSegForCausalLM(Qwen2ForCausalLM, MLCDSegMetaForCausalLM):
                     original_sizes=[(mask_h, mask_w)],
                 )
             pred_mask = net_out[0]
-            return pred_mask
+            mask_tensor = (pred_mask > 0).int()
+            return mask_tensor
         else:
             return torch.zeros([0] + list(image_np.shape[:2]), device=self.device)