Update app.py

app.py

@@ -21,11 +21,13 @@ def estimate_normal_map(image):
     return (normal_map * 255).astype(np.uint8)
 
 
-# Function to apply Thin Plate Spline (TPS) warping
 def apply_tps_warping(design, normal_map):
+    # Resize normal map to match the design size
+    normal_map = cv2.resize(normal_map, (design.shape[1], design.shape[0]))
+
     h, w = design.shape[:2]
     x, y = np.meshgrid(np.arange(w), np.arange(h))
-    
+
     # Generate warp offsets from normal map
     control_x = x + (normal_map[:, :, 0] - 128) * 0.5
     control_y = y + (normal_map[:, :, 1] - 128) * 0.5