Update app.py

app.py

@@ -52,6 +52,10 @@ def stitch_rgbd_videos(
 
             depth_norm = np.clip(depth_norm, 0, 255)
             
+            # Ensure depth_norm is 2D (remove singleton dimensions if present)
+            if depth_norm.ndim == 3:
+                depth_norm = np.squeeze(depth_norm)
+                
             # Generate depth visualization:
             if grayscale:
                 if convert_from_color: