Update app.py

app.py

@@ -19,21 +19,22 @@ title = '''# 👋🏻Welcome to Tonic's🌟🎥StableVideo XT-1-1
 Join us : 🌟TeamTonic🌟 is always making cool demos! Join our active builder's🛠️community 👻  [![Join us on Discord](https://img.shields.io/discord/1109943800132010065?label=Discord&logo=discord&style=flat-square)](https://discord.gg/GWpVpekp) On 🤗Huggingface: [TeamTonic](https://huggingface.co/TeamTonic) & [MultiTransformer](https://huggingface.co/MultiTransformer) On 🌐Github: [Tonic-AI](https://github.com/tonic-ai) & contribute to 🌟 [SciTonic](https://github.com/Tonic-AI/scitonic) 🤗Big thanks to Yuvi Sharma and all the folks at huggingface for the community grant 🤗
 '''
 
-cache_dir = os.path.join(os.path.expanduser("~"), ".cache/huggingface/diffusers")
+script_dir = os.path.dirname(os.path.abspath(__file__))
+downloaded_safetensors_path = os.path.join(script_dir, "svd_xt_1_1.safetensors")
+if not os.path.exists(downloaded_safetensors_path):
+    raise FileNotFoundError(f"The file {downloaded_safetensors_path} was not found.")
 
 original_model_id = "stabilityai/stable-video-diffusion-img2vid-xt"
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     original_model_id,
     torch_dtype=torch.float16,
-    variant="fp16"
+    variant="fp16",
+    use_auth_token=hf_token
 )
 
-# Construct the model cache directory path
+cache_dir = os.path.join(os.path.expanduser("~"), ".cache/huggingface/diffusers")
 model_cache_dir = os.path.join(cache_dir, original_model_id.replace("/", "--"))
-
-downloaded_safetensors_path = "./svd_xt_1_1.safetensors"
-
-original_safetensors_path = os.path.join(model_cache_dir, "./svd_xt.safetensors")
+original_safetensors_path = os.path.join(model_cache_dir, "svd_xt.safetensors")
 os.replace(downloaded_safetensors_path, original_safetensors_path)
 
 pipe = StableVideoDiffusionPipeline.from_pretrained(
@@ -43,8 +44,6 @@ pipe = StableVideoDiffusionPipeline.from_pretrained(
 )
 pipe.to("cuda")
 pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
-#pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=True)
-
 max_64_bit_int = 2**63 - 1
 
 @spaces.GPU(enable_queue=True)