update app

app.py

@@ -18,11 +18,9 @@ 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)
 
 
@@ -39,6 +37,7 @@ def infer(prompt):
     if output_size == "576x1024":
         width = 1024
         height = 576
+        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)
@@ -50,6 +49,7 @@ def infer(prompt):
     elif output_size == "256x256":
         width = 256 
         height = 256
+        config_256 = "configs/inference_t2v_tconv256_v1.0_freenoise.yaml"
         config_256 = OmegaConf.load(config_256)
         model_config_256 = config_256.pop("model", OmegaConf.create())
         model_256 = instantiate_from_config(model_config_256)