Update app.py

app.py

@@ -208,6 +208,11 @@ def infer(path_input, seed=None):
     # Get intensity map from original image
     intensity_map = get_image_intensity(input_array, gamma_correction=1.0)
 
+    # Resize intensity_map to match depth_array dimensions
+    depth_height, depth_width = depth_array.shape[:2]
+    if intensity_map.shape[:2] != (depth_height, depth_width):
+        intensity_map = cv2.resize(intensity_map, (depth_width, depth_height), interpolation=cv2.INTER_LINEAR)
+
     # Blend depth with intensity map
     blended_result = blend_numpy_images(
         cv2.cvtColor(depth_array, cv2.COLOR_RGB2BGR if len(depth_array.shape) == 3 else cv2.COLOR_GRAY2BGR),