Update app.py

app.py

@@ -26,17 +26,16 @@ def segment_dress(image):
 def warp_design(design, mask, warp_scale):
     """Warp the design using TPS and scale control."""
     h, w = mask.shape[:2]
-    
+
     # Resize design to match mask dimensions
     design_resized = cv2.resize(design, (w, h))
 
+    # Convert boolean mask to uint8 (0-255)
+    mask = (mask * 255).astype(np.uint8)
+
     # Ensure mask is single-channel grayscale
     if len(mask.shape) == 3:
         mask = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY)
-    
-    # Ensure mask is uint8
-    if mask.dtype != np.uint8:
-        mask = (mask * 255).astype(np.uint8)
 
     # Convert design_resized to 3-channel if it's grayscale
     if len(design_resized.shape) == 2: