Update app.py

app.py

@@ -15,13 +15,27 @@ from lvdm.utils.common_utils import str2bool
 from lvdm.utils.dist_utils import setup_dist, gather_data
 from lvdm.utils.saving_utils import npz_to_video_grid, npz_to_imgsheet_5d
 from utils import load_model, get_conditions, make_model_input_shape, torch_to_np
-from huggingface_hub import hf_hub_url, cached_download
+from huggingface_hub import hf_hub_download
 
 config_path = "model_config.yaml"
 config = OmegaConf.load(config_path)
 
-REPO_ID = "RamAnanth1/videocrafter-text2video"
-ckpt_path = cached_download(hf_hub_url(REPO_ID, 'model.ckpt'))
+
+# Download model
+REPO_ID = 'VideoCrafter/t2v-version-1-1'
+filename_list = ['models/base_t2v/model.ckpt',
+                'models/videolora/lora_001_Loving_Vincent_style.ckpt',
+                'models/videolora/lora_002_frozenmovie_style.ckpt',
+                'models/videolora/lora_003_MakotoShinkaiYourName_style.ckpt',
+                'models/videolora/lora_004_coco_style.ckpt']
+for filename in filename_list:
+    if not os.path.exists(filename):
+        hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='./', local_dir_use_symlinks=False)
+            
+
+
+ckpt_path = 'models/base_t2v/model.ckpt'
+
 # # get model & sampler
 model, _, _ = load_model(config, ckpt_path, 
                          inject_lora=False,