update

app.py

@@ -1,3 +1,4 @@
+import gc
 import os
 
 import numpy as np
@@ -16,7 +17,11 @@ from huggingface_hub import hf_hub_download
 from depthcrafter.utils import read_video_frames, vis_sequence_depth, save_video
 
 examples = [
-    ["examples/example_01.mp4", 25, 1.2, 1024, 195],
+    ["examples/example_01.mp4", 10, 1.2, 1024, 60],
+    ["examples/example_02.mp4", 10, 1.2, 1024, 60],
+    ["examples/example_03.mp4", 10, 1.2, 1024, 60],
+    ["examples/example_04.mp4", 10, 1.2, 1024, 60],
+    ["examples/example_05.mp4", 10, 1.2, 1024, 60],
 ]
 
 
@@ -83,6 +88,11 @@ def infer_depth(
     save_video(res, save_path + "_depth.mp4", fps=target_fps)
     save_video(vis, save_path + "_vis.mp4", fps=target_fps)
     save_video(frames, save_path + "_input.mp4", fps=target_fps)
+
+    # clear the cache for the next video
+    gc.collect()
+    torch.cuda.empty_cache()
+
     return [
         save_path + "_input.mp4",
         save_path + "_vis.mp4",
@@ -106,7 +116,7 @@ def construct_demo():
                         <a href='https://scholar.google.com/citations?user=4oXBp9UAAAAJ&hl=en'>Ying Shan</a>\
                     </h2> \
                     <a style='font-size:18px;color: #000000'>If you find DepthCrafter useful, please help ⭐ the </a>\
-                    <a style='font-size:18px;color: #FF5DB0' href='https://github.com/wbhu/DepthCrafter'>[Github Repo]</a>\
+                    <a style='font-size:18px;color: #FF5DB0' href='https://github.com/Tencent/DepthCrafter'>[Github Repo]</a>\
                     <a style='font-size:18px;color: #000000'>, which is important to Open-Source projects. Thanks!</a>\
                         <a style='font-size:18px;color: #000000' href='https://arxiv.org/abs/2409.02095'> [ArXiv] </a>\
                         <a style='font-size:18px;color: #000000' href='https://depthcrafter.github.io/'> [Project Page] </a> </div>
@@ -145,7 +155,7 @@ def construct_demo():
                             label="num denoising steps",
                             minimum=1,
                             maximum=25,
-                            value=25,
+                            value=10,
                             step=1,
                         )
                         guidance_scale = gr.Slider(
@@ -166,7 +176,7 @@ def construct_demo():
                             label="process length",
                             minimum=1,
                             maximum=280,
-                            value=195,
+                            value=60,
                             step=1,
                         )
                     generate_btn = gr.Button("Generate")
@@ -184,7 +194,18 @@ def construct_demo():
             ],
             outputs=[output_video_1, output_video_2],
             fn=infer_depth,
-            cache_examples=False,
+            cache_examples=True,
+        )
+        gr.Markdown(
+            """
+            <span style='font-size:18px;color: #E7CCCC'>Note: 
+            For time quota consideration, we set the default parameters to be more efficient here,
+            with a trade-off of shorter video length and slightly lower quality.
+            You may adjust the parameters according to our 
+            <a style='font-size:18px;color: #FF5DB0' href='https://github.com/Tencent/DepthCrafter'>[Github Repo]</a>
+             for better results if you have enough time quota.
+            </span>
+            """
         )
 
         generate_btn.click(

examples/example_01.mp4

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:afb78decc210225793b20d5bca5b13da07c97233e6fabea44bf02eba8a52bdaf
-size 14393250
+oid sha256:99252cdff428d17b2f3348f4b5ab8951a209f87dd1a2cd924c5d4e257825fdc3
+size 803640

examples/example_02.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cadcf18e00165d10aeac49b4d86f3dfb876b4cea07c1717f5f56db5160dbc2c
+size 5714915

examples/example_03.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a7b7f118a1b5b0642cdeb23a3ba7fbff917c42f50312c752d166738de578c586
+size 447055

examples/example_04.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d91a3fc591e93eb31b640f81d8ed1ff051d4a66bd1a38b36f2cf7e9ebcbd9073
+size 1171671

examples/example_05.mp4

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c5625e308eb9a2bc2332f38c936e5bd4b5aeff70d6527aa8005a9a167d3c08db
+size 2324988