Update app.py

app.py

@@ -5,6 +5,7 @@ import os
 from glob import glob
 from pathlib import Path
 from typing import Optional
+from huggingface_hub import HfFolder
 from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import load_image, export_to_video
 from PIL import Image
@@ -18,9 +19,18 @@ 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 🤗
 '''
 
+# Load the API token from an environment variable
+hf_token = os.getenv("HF_TOKEN")
+
+# If the token is not found, raise an error or handle it appropriately
+if not hf_token:
+    raise ValueError("Hugging Face token not found. Please set the HF_TOKEN environment variable.")
+
+# Use the token for authentication
+HfFolder.save_token(hf_token)
 
 pipe = StableVideoDiffusionPipeline.from_pretrained(
-    "stabilityai/stable-video-diffusion-img2vid-xt-1-1", torch_dtype=torch.float16, variant="fp16"
+    "stabilityai/stable-video-diffusion-img2vid-xt-1-1", torch_dtype=torch.float16, variant="fp16", use_auth_token=hf_token
 )
 pipe.to("cuda")
 pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)