Update app.py

app.py

@@ -23,6 +23,15 @@ import PIL
 import torch.nn.functional as F
 from torchvision import transforms
 
+import subprocess
+
+def download_model():
+    command = "huggingface-cli download --resume-download THUDM/CogVideoX-5b-I2V --local-dir ./cogvideox-5b-i2v"
+    process = subprocess.run(command, shell=True, check=True)
+    print("Download completed")
+
+download_model()
+
 def get_prompt(file:str):
     with open(file,'r') as f:
         a=f.readlines()