Update app.py

app.py

@@ -6,18 +6,18 @@ import tempfile
 import spaces
 
 @spaces.GPU
-def initialize_pipeline():
-    # Check if CUDA is available and set the device
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-
+def initialize_pipeline(device):
     # Initialize the pipeline with CUDA support
     pipeline = I2VGenXLPipeline.from_pretrained("ali-vilab/i2vgen-xl", torch_dtype=torch.float16, variant="fp16")
     pipeline.to(device)
-    return pipeline, device
+    return pipeline
 
 def generate_gif(prompt, image, negative_prompt, num_inference_steps, guidance_scale, seed):
-    # Initialize the pipeline and device within the function
-    pipeline, device = initialize_pipeline()
+    # Check if CUDA is available and set the device
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+
+    # Initialize the pipeline within the function
+    pipeline = initialize_pipeline(device)
 
     # Set the generator seed
     generator = torch.Generator(device=device).manual_seed(seed)