Update app.py

app.py

@@ -1,3 +1,6 @@
+
+import spaces
+
 import gradio as gr
 import torch
 from diffusers import DiffusionPipeline
@@ -31,6 +34,7 @@ torch._dynamo.config.capture_dynamic_output_shape_ops = True
 pipe.transformer.compile()
 
 # Gradio inference function
+@spaces.GPU(duration=20)
 def generate_video(prompt, seed):
     generator = torch.manual_seed(seed) if seed else None