Update app.py

app.py

@@ -212,7 +212,15 @@ def dummy(img):
     alpha_channel = img["layers"][0][:, :, 3]
     mask = np.where(alpha_channel == 0, 0, 255).astype(np.uint8)
 
-    return mask
+    # 将蒙版转换为与原始图像相同分辨率的图像
+    mask_image_pil = Image.fromarray(mask)
+    mask_image_resized = mask_image_pil.resize((composite_width, composite_height), Image.ANTIALIAS)
+
+    # 将调整大小后的蒙版图像转换回 numpy 数组
+    mask_resized = np.array(mask_image_resized)
+
+    return mask_resized
+
 
 
 # Function to update the image in Tab 1