add gpu annotation to load model function

musepose_inference.py

@@ -191,6 +191,7 @@ class MusePoseInference:
         )
         return output_path, output_path_demo
 
+    @spaces.GPU(duration=120)
     def init_model(self,
                    weight_dtype: torch.dtype,
                    infer_config: DictConfig

pose_align.py

@@ -312,6 +312,7 @@ class PoseAlignmentInference:
         print('pose align done')
         return outfn_align_pose_video, outfn
 
+    @spaces.GPU(duration=120)
     def init_model(self):
         if self.detector is None:
             device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')