Spaces:
Running
on
Zero
Running
on
Zero
Commit
Β·
4b7cf01
1
Parent(s):
fc6739e
Add GPU decorator to complete_pipeline method in UniRigDemo for enhanced performance
Browse files
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.
|