Update app.py

app.py

@@ -27,14 +27,23 @@ from models.RIFE.IFNet_HDv3 import IFNet
 hf_token = os.environ.get('HF_TOKEN')
 device = "cuda"
 
+model_ids = [
+    'runwayml/stable-diffusion-v1-5',
+    'lllyasviel/sd-controlnet-depth', 
+    'lllyasviel/sd-controlnet-canny', 
+    'lllyasviel/sd-controlnet-openpose',
+]
+for model_id in model_ids:
+    model_name = model_id.split('/')[-1]
+    snapshot_download(model_id, local_dir=f'checkpoints/{model_name}')
+
+sd_path = "checkpoints/stable-diffusion-v1-5"
 
-sd_path = "runwayml/stable-diffusion-v1-5"
-snapshot_download(sd_path, local_dir="checkpoints")
 inter_path = "checkpoints/flownet.pkl"
 controlnet_dict = {
-    "pose": "lllyasviel/sd-controlnet-openpose",
-    "depth": "lllyasviel/sd-controlnet-depth",
-    "canny": "lllyasviel/sd-controlnet-canny",
+    "pose": "checkpoints/sd-controlnet-openpose",
+    "depth": "checkpoints/sd-controlnet-depth",
+    "canny": "checkpoints/sd-controlnet-canny",
 }