Update app.py

app.py

@@ -4,8 +4,8 @@ from diffusers import I2VGenXLPipeline
 from diffusers.utils import export_to_gif, load_image
 import tempfile
 
-# Check if CUDA is available and set the device
-device = "cuda" if torch.cuda.is_available() else "cpu"
+def initialize_pipeline():
+    device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Initialize the pipeline with CUDA support
 pipeline = I2VGenXLPipeline.from_pretrained("ali-vilab/i2vgen-xl", torch_dtype=torch.float16, variant="fp16")