update

app.py

@@ -84,6 +84,7 @@ def _load_pipeline(config_path: str, checkpoint_path: Optional[str], use_ema: bo
 def build_predict(config_path: str, checkpoint_path: Optional[str], output_dir: str, use_ema: bool):
     os.makedirs(output_dir, exist_ok=True)
 
+    @spaces.GPU 
     def predict(prompt: str, num_frames: int) -> str:
         if not prompt or not prompt.strip():
             raise gr.Error("Please enter a non-empty text prompt.")