Update app.py
Browse files
app.py
CHANGED
@@ -32,7 +32,7 @@ def warp_design(design, mask, warp_scale):
|
|
32 |
scaled_mask = (mask * 255 * (warp_scale / 100)).astype(np.uint8)
|
33 |
|
34 |
# Ensure mask is single-channel
|
35 |
-
if len(scaled_mask.shape) == 3:
|
36 |
scaled_mask = cv2.cvtColor(scaled_mask, cv2.COLOR_BGR2GRAY)
|
37 |
|
38 |
return cv2.bitwise_and(design_resized, design_resized, mask=scaled_mask)
|
|
|
32 |
scaled_mask = (mask * 255 * (warp_scale / 100)).astype(np.uint8)
|
33 |
|
34 |
# Ensure mask is single-channel
|
35 |
+
if len(scaled_mask.shape) == 3 and scaled_mask.shape[2] == 3:
|
36 |
scaled_mask = cv2.cvtColor(scaled_mask, cv2.COLOR_BGR2GRAY)
|
37 |
|
38 |
return cv2.bitwise_and(design_resized, design_resized, mask=scaled_mask)
|