Update app.py

app.py

@@ -21,15 +21,12 @@ Join us : 🌟TeamTonic🌟 is always making cool demos! Join our active builder
 
 # 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)
 
-# Load the original model to cache
+cache_dir = os.path.join(os.path.expanduser("~"), ".cache/huggingface/diffusers")
 original_model_id = "stabilityai/stable-video-diffusion-img2vid-xt"
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     original_model_id,
@@ -38,16 +35,15 @@ pipe = StableVideoDiffusionPipeline.from_pretrained(
     use_auth_token=hf_token
 )
 
-# Get the cache directory of the original model
-cache_dir = os.path.join(os.path.expanduser("~"), ".cache/huggingface/diffusers")
+# Construct the model cache directory path
 model_cache_dir = os.path.join(cache_dir, original_model_id.replace("/", "--"))
 
+# Path to the downloaded svd_xt_1_1.safetensors file
 downloaded_safetensors_path = "svd_xt_1_1.safetensors"
 
 original_safetensors_path = os.path.join(model_cache_dir, "svd_xt.safetensors")
 os.replace(downloaded_safetensors_path, original_safetensors_path)
 
-# Load the model from the local directory with the replaced safetensors file
 pipe = StableVideoDiffusionPipeline.from_pretrained(
     model_cache_dir,
     torch_dtype=torch.float16,