Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import sys
 import time
@@ -45,7 +46,7 @@ def init_predictor(task_type: str):
     except Exception as e:
         return f"Error loading model: {e}"
 
-
+@spaces.GPU(duration=90)
 def generate_video(prompt, seed, image=None, task_type=None):
     global predictor