Update app.py

app.py

@@ -5,6 +5,7 @@ import sys
 import tempfile
 import shutil
 import subprocess
+import spaces
 
 # --- Configuration ---
 # Path to the cloned UniRig repository directory within the Space
@@ -29,6 +30,7 @@ if DEVICE.type == 'cuda':
 else:
     print("Warning: CUDA not available or not detected by PyTorch. UniRig performance will be severely impacted.")
 
+@spaces.GPU
 def run_unirig_command(command_args, step_name):
     """Helper function to run UniRig commands using subprocess."""
     python_exe = sys.executable  # Use the current python interpreter
@@ -66,6 +68,7 @@ def run_unirig_command(command_args, step_name):
 
 
 # --- Core Rigging Function ---
+@spaces.GPU
 def rig_glb_mesh_multistep(input_glb_file_obj):
     """
     Takes an input GLB file object, rigs it using the new UniRig multi-step process,