MohamedRashad commited on
Commit
4b7cf01
Β·
1 Parent(s): fc6739e

Add GPU decorator to complete_pipeline method in UniRigDemo for enhanced performance

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,6 @@ class UniRigDemo:
50
  file_ext = Path(file_path).suffix.lower()
51
  return file_ext in self.supported_formats
52
 
53
- @spaces.GPU()
54
  def generate_skeleton(self, input_file: str, seed: int = 12345) -> Tuple[str, str, str]:
55
  """
56
  OPERATION 1: Generate skeleton for the input 3D model using Python
@@ -130,6 +129,7 @@ class UniRigDemo:
130
 
131
  return "βœ… Model rigging completed successfully!", str(output_file_abs), preview_info
132
 
 
133
  def complete_pipeline(self, input_file: str, seed: int = 12345) -> Tuple[str, str, str, str, str]:
134
  """
135
  Run the complete rigging pipeline: skeleton generation β†’ skinning β†’ merge.
 
50
  file_ext = Path(file_path).suffix.lower()
51
  return file_ext in self.supported_formats
52
 
 
53
  def generate_skeleton(self, input_file: str, seed: int = 12345) -> Tuple[str, str, str]:
54
  """
55
  OPERATION 1: Generate skeleton for the input 3D model using Python
 
129
 
130
  return "βœ… Model rigging completed successfully!", str(output_file_abs), preview_info
131
 
132
+ @spaces.GPU()
133
  def complete_pipeline(self, input_file: str, seed: int = 12345) -> Tuple[str, str, str, str, str]:
134
  """
135
  Run the complete rigging pipeline: skeleton generation β†’ skinning β†’ merge.