update app

app.py

@@ -17,6 +17,15 @@ from funcs import (
 )
 from utils.utils import instantiate_from_config
 
+ckpt_path_1024 = "checkpoints/base_1024_v1/model.ckpt"
+config_1024 = "configs/inference_t2v_1024_v1.0_freenoise.yaml"
+# hf_hub_download(repo_id="VideoCrafter/Text2Video-1024", filename="model.ckpt", local_dir=ckpt_path_1024)
+
+ckpt_path_256 = "checkpoints/base_256_v1/model.pth"
+config_256 = "configs/inference_t2v_tconv256_v1.0_freenoise.yaml"
+hf_hub_download(repo_id="MoonQiu/LongerCrafter", filename="model.pth", local_dir=ckpt_path_256)
+
+
 def infer(prompt):
     output_size = "256x256"
     num_frames = 32 
@@ -30,9 +39,6 @@ def infer(prompt):
     if output_size == "576x1024":
         width = 1024
         height = 576
-        ckpt_path_1024 = "checkpoints/base_1024_v1/model.ckpt"
-        hf_hub_download(repo_id="VideoCrafter/Text2Video-1024", filename="model.ckpt", local_dir=ckpt_path_1024)
-        config_1024 = "configs/inference_t2v_1024_v1.0_freenoise.yaml"
         config_1024 = OmegaConf.load(config_1024)
         model_config_1024 = config_1024.pop("model", OmegaConf.create())
         model_1024 = instantiate_from_config(model_config_1024)
@@ -44,9 +50,6 @@ def infer(prompt):
     elif output_size == "256x256":
         width = 256 
         height = 256
-        ckpt_path_256 = "checkpoints/base_256_v1/model.pth"
-        config_256 = "configs/inference_t2v_tconv256_v1.0_freenoise.yaml"
-        hf_hub_download(repo_id="MoonQiu/LongerCrafter", filename="model.pth", local_dir=model_config_256)
         config_256 = OmegaConf.load(config_256)
         model_config_256 = config_256.pop("model", OmegaConf.create())
         model_256 = instantiate_from_config(model_config_256)