GPU setting

gradio_app.py

@@ -12,6 +12,7 @@ HF_HOME = "/home/user/app/hf_cache"
 os.environ["HF_HOME"] = HF_HOME
 os.environ["TRANSFORMERS_CACHE"] = HF_HOME
 os.makedirs(HF_HOME, exist_ok=True)
+from spaces import GPU
 
 # hf_hub_download(repo_id="ai-forever/Real-ESRGAN", filename="RealESRGAN_x4.pth", local_dir="model_real_esran")
 # snapshot_download(repo_id="AlexWortega/RIFE", local_dir="model_rife")
@@ -39,6 +40,7 @@ def download_models():
 # -----------------------------
 # Step 1: Get Anchor Video
 # -----------------------------
+@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
 # -----------------------------
+@GPU
 def inference(
     fps, num_frames, controlnet_weights, controlnet_guidance_start,
     controlnet_guidance_end, guidance_scale, num_inference_steps, dtype,