adding spaces.gpu

gradio_app.py

@@ -5,6 +5,7 @@ from pathlib import Path
 import gradio as gr
 import numpy as np
 import os
+import spaces
 # -----------------------------
 # Setup paths and env
 # -----------------------------
@@ -39,6 +40,7 @@ def download_models():
 # -----------------------------
 # Step 1: Get Anchor Video
 # -----------------------------
+@spaces.GPU
 def get_anchor_video(video_path, fps, num_frames, target_pose, mode,
                        radius_scale, near_far_estimated,
                        sampler_name, diffusion_guidance_scale, diffusion_inference_steps,
@@ -110,6 +112,7 @@ def get_anchor_video(video_path, fps, num_frames, target_pose, mode,
 # -----------------------------
 # Step 2: Run Inference
 # -----------------------------
+@spaces.GPU
 def inference(
     fps, num_frames, controlnet_weights, controlnet_guidance_start,
     controlnet_guidance_end, guidance_scale, num_inference_steps, dtype,