Update app.py

app.py

@@ -177,7 +177,12 @@ def laplacian_blend(img1: np.ndarray, img2: np.ndarray, mask: np.ndarray, depth:
     imgs = []
     for d in range(0, depth-1):
         gaussian_img = _low_pass_filter(blanded_img.copy(), sigma)
-        gaussian_img = cv2.resize(gaussian_img, blended[d].shape[:2])  # Ensure sizes match
+        gaussian_img = cv2.resize(gaussian_img, (blended[d].shape[1], blended[d].shape[0]))  # Ensure sizes match
+        
+        # Ensure the number of channels match
+        if len(gaussian_img.shape) == 2:
+            gaussian_img = np.stack([gaussian_img] * 3, axis=-1)
+        
         reconstructed_img = cv2.add(blended[d], gaussian_img)
         
         imgs.append(reconstructed_img)