Update app_t2v.py

app_t2v.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 from diffusers import WanPipeline, AutoencoderKLWan
@@ -19,6 +20,12 @@ pipe = WanPipeline.from_pretrained(
 pipe.to(device)
 
 # Core inference function
+def get_duration(
+   prompt, negative_prompt, height, width, num_frames, guidance_scale, guidance_scale_2, num_steps
+):
+    return steps * 15
+
+@spaces.GPU(duration=get_duration)
 def generate_video(prompt, negative_prompt, height, width, num_frames, guidance_scale, guidance_scale_2, num_steps):
     with torch.autocast("cuda", dtype=dtype):
         output = pipe(