Update app.py

app.py

@@ -7,16 +7,13 @@ import matplotlib.cm as cm
 import matplotlib  # New import for the updated colormap API
 import subprocess
 import sys
-import spaces
 
 from utils.dc_utils import read_video_frames, save_video
 
-title = "#RGBD sbs output"
+title = """#""RGBD SBS output"
 description = """**Video Depth Anything** + RGBD sbs output for viewing with Looking Glass Factory displays.
 Please refer to our [paper](https://arxiv.org/abs/2501.12375), [project page](https://videodepthanything.github.io/), and [github](https://github.com/DepthAnything/Video-Depth-Anything) for more details."""
 
-@spaces.GPU(enable_queue=True)
-
 def stitch_rgbd_videos(
     processed_video: str,
     depth_vis_video: str,
@@ -58,7 +55,12 @@ def stitch_rgbd_videos(
                     # then convert that color image to grayscale.
                     cmap = matplotlib.colormaps.get_cmap("inferno")
                     depth_color = (cmap(depth_norm / 255.0)[..., :3] * 255).astype(np.uint8)
-                    depth_gray = cv2.cvtColor(depth_color, cv2.COLOR_RGB2GRAY)
+                    
+                    if len(depth_color.shape) == 3 and depth_color.shape[2] in [3, 4]:
+                        depth_gray = cv2.cvtColor(depth_color, cv2.COLOR_RGB2GRAY)
+                    else:
+                        depth_gray = depth_color
+                    
                     depth_vis = np.stack([depth_gray] * 3, axis=-1)
                 else:
                     # Directly generate a grayscale image from the normalized depth values.