Update app.py

app.py

@@ -56,7 +56,7 @@ def process_image(img):
     # 将 mask 转换为 PIL 图像（使用对应颜色的 RGB 值）
     mask_images = []
     for i, mask in enumerate(masks):
-        mask_array = np.zeros((height, width, 3), dtype=np.uint8)  # RGB 图像
+        mask_array = np.ones((height, width, 3), dtype=np.uint8) * 255  # 白色背景
         mask_array[mask] = palette[i]  # 使用对应颜色填充
         mask_image = Image.fromarray(mask_array, mode='RGB')
         mask_images.append(mask_image)
@@ -120,11 +120,9 @@ with gr.Blocks() as demo:
     2. Use the brush tool to draw masks with different colors.
     3. Click the **Generate Masks** button to create and save the masks.
     4. View the generated masks on the right side.
-
     ## 📂 **Output**
     - Each mask is saved as an RGB image in the `output_masks` directory.
     - A merged mask is also generated for your convenience.
-
     ## 🚀 **Enjoy!**
     """)