update

app.py

@@ -8,6 +8,8 @@ import random
 from inference import get_parser
 from datetime import datetime
 import argparse
+import spaces #fixme
+from huggingface_hub import snapshot_download
 
 # 解析命令行参数
 
@@ -28,6 +30,15 @@ img_examples = [
 
 max_seed = 2 ** 31
 
+os.makedirs('./checkpoints/',exist_ok=True)
+def download_model():
+    snapshot_download(repo_id="TrajectoryCrafter/TrajectoryCrafter", local_dir="checkpoints/TrajectoryCrafter", local_dir_use_symlinks=False)
+    snapshot_download(repo_id="tencent/DepthCrafter", local_dir="checkpoints/DepthCrafter", local_dir_use_symlinks=False)
+    snapshot_download(repo_id="stabilityai/stable-video-diffusion-img2vid", local_dir="checkpoints/stable-video-diffusion-img2vid", local_dir_use_symlinks=False)
+    snapshot_download(repo_id="alibaba-pai/CogVideoX-Fun-V1.1-5b-InP", local_dir="checkpoints/CogVideoX-Fun-V1.1-5b-InP", local_dir_use_symlinks=False)
+    snapshot_download(repo_id="Salesforce/blip2-opt-2.7b", local_dir="checkpoints/blip2-opt-2.7b", local_dir_use_symlinks=False)
+download_model() #fixme
+
 parser = get_parser() # infer_config.py
 opts = parser.parse_args() # default device: 'cuda:0'
 opts.weight_dtype = torch.bfloat16
@@ -37,6 +48,8 @@ os.makedirs(opts.save_dir,exist_ok=True)
 test_tensor = torch.Tensor([0]).cuda()
 opts.device = str(test_tensor.device)
 
+
+
 CAMERA_MOTION_MODE = ["Basic Camera Trajectory", "Custom Camera Trajectory"]
 
 def show_traj(mode):
@@ -84,14 +97,13 @@ def trajcrafter_demo(opts):
     }
     """
     image2video = TrajCrafter(opts,gradio=True)
-    # image2video.run_both = spaces.GPU(image2video.run_both, duration=290) # fixme
+    image2video.run_gradio = spaces.GPU(image2video.run_gradio, duration=300) # fixme
     with gr.Blocks(analytics_enabled=False, css=css) as trajcrafter_iface:
-        gr.Markdown("<div align='center'> <h1> TrajectoryCrafter: Redirecting View Trajectory for Monocular Videos via Diffusion Models </span> </h1>")
-        #             #   <h2 style='font-weight: 450; font-size: 1rem; margin: 0rem'>\
-        #             #  <a style='font-size:18px;color: #000000' href='https://arxiv.org/abs/2409.02048'> [ArXiv] </a>\
-        #             #  <a style='font-size:18px;color: #000000' href='https://drexubery.github.io/ViewCrafter/'> [Project Page] </a>\
-        #             #  <a style='font-size:18px;color: #FF5DB0' href='https://github.com/Drexubery/ViewCrafter'> [Github] </a>\
-        #             #  <a style='font-size:18px;color: #000000' href='https://www.youtube.com/watch?v=WGIEmu9eXmU'> [Video] </a> </div>")
+        gr.Markdown("<div align='center'> <h1> TrajectoryCrafter: Redirecting View Trajectory for Monocular Videos via Diffusion Models </span> </h1>
+                    <a style='font-size:18px;color: #FF5DB0' href='https://github.com/TrajectoryCrafter/TrajectoryCrafter'> [Github] </a>\                  
+                    #  <a style='font-size:18px;color: #000000' href='https://arxiv.org/abs/2409.02048'> [ArXiv] </a>\
+                     <a style='font-size:18px;color: #000000' href='https://trajectorycrafter.github.io/'> [Project Page] </a>\
+                     <a style='font-size:18px;color: #000000' href='https://www.youtube.com/watch?v=dQtHFgyrids'> [Video] </a> </div>")
 
 
         with gr.Row(equal_height=True):
@@ -277,8 +289,8 @@ def trajcrafter_demo(opts):
 
 trajcrafter_iface = trajcrafter_demo(opts)
 trajcrafter_iface.queue(max_size=10)
-# trajcrafter_iface.launch(server_name=args.server_name, max_threads=10, debug=True)
-trajcrafter_iface.launch(server_name="0.0.0.0", server_port=12345, debug=True, share=False, max_threads=10)
+trajcrafter_iface.launch()
+# trajcrafter_iface.launch(server_name="0.0.0.0", server_port=12345, debug=True, share=False, max_threads=10)