Update app.py

app.py

@@ -5,6 +5,7 @@ import cv2
 import gradio as gr
 import numpy as np
 import matplotlib.cm as cm
+import matplotlib  # New import for the updated colormap API
 import subprocess
 
 from video_depth_anything.video_depth import VideoDepthAnything
@@ -101,17 +102,18 @@ def infer_video_depth(
             # Generate depth visualization:
             if grayscale:
                 if convert_from_color:
-                    # Generate a color depth image first, then convert it to grayscale.
-                    cmap = cm.get_cmap("inferno")
+                    # First, generate a color depth image using the inferno colormap,
+                    # 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)
                     depth_vis = np.stack([depth_gray] * 3, axis=-1)
                 else:
-                    # Directly generate a grayscale image from normalized depth.
+                    # Directly generate a grayscale image from the normalized depth values.
                     depth_vis = np.stack([depth_norm] * 3, axis=-1)
             else:
                 # Generate a color depth image using the inferno colormap.
-                cmap = cm.get_cmap("inferno")
+                cmap = matplotlib.colormaps.get_cmap("inferno")
                 depth_vis = (cmap(depth_norm / 255.0)[..., :3] * 255).astype(np.uint8)
             # Apply Gaussian blur if requested.
             if blur > 0:
@@ -189,7 +191,7 @@ def construct_demo():
             outputs=[processed_video, depth_vis_video, stitched_video],
             fn=infer_video_depth,
             cache_examples=True,
-            cache_mode="sync",
+            cache_mode="lazy",
         )
         
         generate_btn.click(