GPU setting

gradio_app.py

@@ -5,6 +5,8 @@ from pathlib import Path
 import gradio as gr
 import numpy as np
 import os
+from huggingface_hub import spaces
+
 # -----------------------------
 # Setup paths and env
 # -----------------------------
@@ -12,7 +14,6 @@ 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")
@@ -40,7 +41,7 @@ def download_models():
 # -----------------------------
 # Step 1: Get Anchor Video
 # -----------------------------
-@GPU
+@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,
@@ -112,7 +113,7 @@ def get_anchor_video(video_path, fps, num_frames, target_pose, mode,
 # -----------------------------
 # Step 2: Run Inference
 # -----------------------------
-@GPU
+@spaces.GPU
 def inference(
     fps, num_frames, controlnet_weights, controlnet_guidance_start,
     controlnet_guidance_end, guidance_scale, num_inference_steps, dtype,