Update app.py

app.py

@@ -32,7 +32,14 @@ def download_model():
     process = subprocess.run(command, shell=True, check=True)
     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)
+    print("Download completed")
+    
 download_model()
+download_model_senorita()
 
 print("Download successfully!")
 
@@ -87,7 +94,7 @@ def init_pipe():
     control_config['control_in_channels'] = 16
     controlnet_transformer = ControlCogVideoXTransformer3DModel(**control_config)
 
-    all_state_dicts = torch.load("./models_half/ff_controlnet_half.pth", map_location="cpu",weights_only=True)
+    all_state_dicts = torch.load("./senorita-2m/models_half/ff_controlnet_half.pth", map_location="cpu",weights_only=True)
     transformer_state_dict = unwarp_model(all_state_dicts["transformer_state_dict"])
     controlnet_transformer_state_dict = unwarp_model(all_state_dicts["controlnet_transformer_state_dict"])