gaur3009 commited on
Commit
5cd8a4a
·
verified ·
1 Parent(s): ef61ae7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)