Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import os
 import gradio as gr
 import torch
@@ -79,6 +80,7 @@ except Exception as e:
 # ============================================================
 # 5️⃣ Generation function
 # ============================================================
+@spaces.GPU(duration = 40)
 def generate_output(prompt, mode, duration, width, height, steps, guidance, scheduler):
     if pipe is None:
         return None, "❌ Pipeline not initialized."