Update app.py

app.py

@@ -18,8 +18,10 @@ import random
 from huggingface_hub import login, hf_hub_download
 import spaces
 
+model_directory = './checkpoints'
+
 try:
-    hf_hub_download(repo_id="vdo/stable-video-diffusion-img2vid-xt-1-1", local_dir="./checkpoints", cache_dir="./checkpoints")
+    hf_hub_download(repo_id="vdo/stable-video-diffusion-img2vid-xt-1-1", filename="svd_xt_1_1.safetensors", local_dir=model_directory, cache_dir=model_directory)
 except (Exception, BaseException) as error:
     print(error)
 
@@ -32,8 +34,6 @@ except (Exception, BaseException) as error:
 
 # pipe.save_pretrained("./checkpoints", variant="fp16")
 
-model_directory = './checkpoints'
-
 if not os.path.exists(model_directory):
     pipe = StableVideoDiffusionPipeline.from_pretrained(
         # "stabilityai/stable-video-diffusion-img2vid-xt-1-1",