Update app.py

app.py

@@ -26,19 +26,18 @@ from torchvision import transforms
 
 import subprocess
 import spaces
+from huggingface_hub import snapshot_download
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 
 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)
+    snapshot_download(repo_id="THUDM/CogVideoX-5b-I2V", local_dir="./cogvideox-5b-i2v")
     print("Download completed")
 
 
 def download_model_senorita():
-    command = "huggingface-cli download --resume-download PengWeixuanSZU/Senorita-2M --local-dir ./senorita-2m"
-    process = subprocess.run(command, shell=True, check=True)
+    snapshot_download(repo_id="PengWeixuanSZU/Senorita-2M", local_dir="./senorita-2m")
     print("Download completed")
     
 download_model()