Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import os
 import subprocess
 import tempfile
@@ -11,6 +12,7 @@ ckpt_dir = snapshot_download(repo_id, local_dir_use_symlinks=False)
 print(f"Using checkpoints from {ckpt_dir}")
 
 # ---------------- Step 2: Generation Functions ----------------
+@spaces.GPU(duration=100)
 def generate_t2v(prompt, size="1280*704"):
     """Text-to-Video generation."""
     if not prompt.strip():
@@ -45,6 +47,8 @@ def generate_t2v(prompt, size="1280*704"):
 
     return output_path, "Text-to-Video generated successfully!"
 
+
+@spaces.GPU(duration=100)    
 def generate_i2v(image, prompt, size="1280*704"):
     """Image-to-Video generation."""
     if image is None or not prompt.strip():