Update app.py

app.py

@@ -5,7 +5,6 @@ from diffusers.utils import export_to_gif, load_image
 import tempfile
 import spaces
 
-# Function to initialize the pipeline with CUDA support
 @spaces.GPU
 def initialize_pipeline():
     # Check if CUDA is available and set the device
@@ -16,7 +15,10 @@ def initialize_pipeline():
     pipeline.to(device)
     return pipeline, device
 
-def generate_gif(pipeline, device, prompt, image, negative_prompt, num_inference_steps, guidance_scale, seed):
+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()
+
     # Set the generator seed
     generator = torch.Generator(device=device).manual_seed(seed)
 
@@ -49,8 +51,6 @@ def generate_gif(pipeline, device, prompt, image, negative_prompt, num_inference
 
 # Create the Gradio interface with tabs
 with gr.Blocks() as demo:
-    pipeline, device = initialize_pipeline()
-
     with gr.TabItem("Generate from Text or Image"):
         with gr.Row():
             with gr.Column():
@@ -67,7 +67,7 @@ with gr.Blocks() as demo:
 
         generate_button.click(
             fn=generate_gif,
-            inputs=[pipeline, device, prompt, image, negative_prompt, num_inference_steps, guidance_scale, seed],
+            inputs=[prompt, image, negative_prompt, num_inference_steps, guidance_scale, seed],
             outputs=output_video
         )