Update app.py

app.py

@@ -26,13 +26,20 @@ import warnings
 print("gr file", gr.__file__)
 
 from huggingface_hub import hf_hub_download, snapshot_download
+
 os.makedirs("checkpoints", exist_ok=True)
+
 snapshot_download(
     "wwen1997/framer_512x320",
     local_dir="checkpoints/framer_512x320",
     token=os.environ["TOKEN"],
 )
 
+snapshot_download(
+    "stabilityai/stable-video-diffusion-img2vid-xt",
+    local_dir="checkpoints/stable-video-diffusion-img2vid-xt",
+    token=os.environ["TOKEN"],
+)
 
 
 def get_args():
@@ -382,7 +389,7 @@ class Drag:
         #         "xformers is not available. Make sure it is installed correctly")
 
         pipe = StableVideoDiffusionInterpControlPipeline.from_pretrained(
-            "stabilityai/stable-video-diffusion-img2vid-xt",
+            "checkpoints/stable-video-diffusion-img2vid-xt",
             unet=unet,
             controlnet=controlnet,
             low_cpu_mem_usage=False,