add spaces annotation

musepose_inference.py

@@ -1,7 +1,6 @@
 import os
 from datetime import datetime
 from pathlib import Path
-
 import torch
 from diffusers import AutoencoderKL, DDIMScheduler
 from einops import repeat
@@ -20,6 +19,9 @@ from musepose.pipelines.pipeline_pose2vid_long import Pose2VideoPipeline
 from musepose.utils.util import get_fps, read_frames, save_videos_grid
 from downloading_weights import download_models
 
+# ZeroGPU
+import spaces
+
 
 class MusePoseInference:
     def __init__(self,
@@ -48,6 +50,7 @@ class MusePoseInference:
         if not os.path.exists(self.output_dir):
             os.makedirs(self.output_dir)
 
+    @spaces.GPU
     def infer_musepose(
         self,
         ref_image_path: str,

pose_align.py

@@ -11,6 +11,9 @@ from pose.script.dwpose import DWposeDetector, draw_pose
 from pose.script.util import size_calculate, warpAffine_kps
 from downloading_weights import download_models
 
+# ZeroGPU
+import spaces
+
 
 '''
     Detect dwpose from img, then align it by scale parameters
@@ -36,6 +39,7 @@ class PoseAlignmentInference:
         if not os.path.exists(self.output_dir):
             os.makedirs(self.output_dir)
 
+    @spaces.GPU
     def align_pose(
         self,
         vidfn: str,