add spaces

app.py

@@ -24,6 +24,12 @@ from utils.utils_freetraj import plan_path
 
 MAX_KEYS = 5
 
+ckpt_dir_512 = "checkpoints/base_512_v2"
+ckpt_path_512 = "checkpoints/base_512_v2/model.ckpt"
+if not os.path.exists(ckpt_path_512):
+    os.makedirs(ckpt_dir_512, exist_ok=True)
+    hf_hub_download(repo_id="VideoCrafter/VideoCrafter2", filename="model.ckpt", local_dir=ckpt_dir_512, force_download=True)
+    
 def check_move(trajectory, video_length=16):
     traj_len = len(trajectory)
     if traj_len < 2:
@@ -40,7 +46,7 @@ def check_move(trajectory, video_length=16):
 
     return True
 
-@spaces.GPU(duration=600)
+@spaces.GPU(duration=180)
 def infer(*user_args):
     prompt_in = user_args[0]
     target_indices = user_args[1]
@@ -62,8 +68,6 @@ def infer(*user_args):
     video_length = 16
     width = 512 
     height = 320
-    ckpt_dir_512 = "checkpoints/base_512_v2"
-    ckpt_path_512 = "checkpoints/base_512_v2/model.ckpt"
     if radio_mode == 'ori':
         config_512 = "configs/inference_t2v_512_v2.0.yaml"
     else:
@@ -98,14 +102,7 @@ def infer(*user_args):
     model_config_512 = config_512.pop("model", OmegaConf.create())
     model = instantiate_from_config(model_config_512)
     model = model.cuda()
-    if not os.path.exists(ckpt_path_512):
-        os.makedirs(ckpt_dir_512, exist_ok=True)
-        hf_hub_download(repo_id="VideoCrafter/VideoCrafter2", filename="model.ckpt", local_dir=ckpt_dir_512)
-    try:
-        model = load_model_checkpoint(model, ckpt_path_512)
-    except:
-        hf_hub_download(repo_id="VideoCrafter/VideoCrafter2", filename="model.ckpt", local_dir=ckpt_dir_512, force_download=True)
-        model = load_model_checkpoint(model, ckpt_path_512)
+    model = load_model_checkpoint(model, ckpt_path_512)
     model.eval()
 
     if seed is None: