release v1 demo

app.py

@@ -38,6 +38,7 @@ import mediapy as media
 
 from huggingface_hub import login
 
+from gradio_patches.examples import Examples
 from chronodepth.unet_chronodepth import DiffusersUNetSpatioTemporalConditionModelChronodepth
 from chronodepth.chronodepth_pipeline import ChronoDepthPipeline
 from chronodepth.video_utils import resize_max_res, colorize_video_depth
@@ -250,16 +251,17 @@ def run_demo_server(pipe):
                     elem_id="download",
                     interactive=False,
                 )
-        gr.Examples(
+        Examples(
+            fn=process_pipe_video,
             examples=[
                 ["files/elephant.mp4"],
                 ["files/kitti360_seq_0000.mp4"],
             ],
             inputs=[video_input],
             outputs=[video_output_video, video_output_files],
-            fn=process_pipe_video,
+
             cache_examples=True,
-            cache_mode="eager",
+            directory_name="examples_video",
         )
 
         video_submit_btn.click(

gradio_patches/examples.py

@@ -0,0 +1,13 @@
+from pathlib import Path
+
+import gradio
+from gradio.utils import get_cache_folder
+
+
+class Examples(gradio.helpers.Examples):
+    def __init__(self, *args, directory_name=None, **kwargs):
+        super().__init__(*args, **kwargs, _initiated_directly=False)
+        if directory_name is not None:
+            self.cached_folder = get_cache_folder() / directory_name
+            self.cached_file = Path(self.cached_folder) / "log.csv"
+        self.create()

requirements.txt

@@ -1,5 +1,5 @@
 spaces
-gradio==5.7.1
+gradio==4.32.1
 diffusers==0.29.1
 easydict==1.13
 einops==0.8.0