Update app_t2v.py

app_t2v.py

@@ -2,6 +2,7 @@ import gradio as gr
 import torch
 from diffusers import WanPipeline, AutoencoderKLWan
 from diffusers.utils import export_to_video
+import spaces
 import os
 import uuid
 
@@ -29,6 +30,7 @@ DEFAULT_NEGATIVE_PROMPT = (
 )
 
 # Inference function
+@spaces.GPU(duration=150)
 def generate_video(prompt, num_frames=81, steps=40, scale=4.0, scale2=3.0):
     if not prompt.strip():
         return "Prompt is empty.", None