Update app.py

app.py

@@ -5,6 +5,7 @@ import sys
 import tempfile
 import shutil
 import subprocess
+import spaces
 # from huggingface_hub import HfApi, snapshot_download # For future model management if needed
 # import spaces # For @spaces.GPU decorator if you add it
 
@@ -24,6 +25,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
@@ -65,9 +67,8 @@ def run_unirig_command(command_args, step_name):
         raise gr.Error(f"Unexpected Python error during {step_name}: {str(e_general)[:500]}")
 
 
-# If you are using @spaces.GPU, you would import it:
-# import spaces
 # @spaces.GPU # You can specify type like @spaces.GPU(type="t4") or count
+@spaces.GPU
 def rig_glb_mesh_multistep(input_glb_file_obj):
     """
     Takes an input GLB file object (from gr.File with type="filepath"),