Update processing.py

processing.py

@@ -6,6 +6,7 @@ import gc
 import shutil
 import sys
 from assets.i18n.i18n import I18nAuto
+import spaces
 
 i18n = I18nAuto()
 current_dir = os.path.dirname(os.path.abspath(__file__))
@@ -107,6 +108,7 @@ def extract_model_name_from_checkpoint(checkpoint_path):
     print(f"Original checkpoint path: {checkpoint_path}, extracted model_name: {model_name}")
     return model_name.strip()
 
+@spaces.GPU
 def run_command_and_process_files(
     model_type,
     config_path,
@@ -289,6 +291,7 @@ def run_command_and_process_files(
         traceback.print_exc()
         return (None,) * 14
 
+@spaces.GPU
 def process_audio(
     input_audio_file,
     model,
@@ -460,6 +463,7 @@ def process_audio(
             update_progress_html("Error occurred", 0)
         )
 
+@spaces.GPU
 def ensemble_audio_fn(files, method, weights, progress=gr.Progress()):
     try:
         if len(files) < 2:
@@ -512,6 +516,7 @@ def ensemble_audio_fn(files, method, weights, progress=gr.Progress()):
         progress(100, desc="Ensemble process completed")
 
 
+@spaces.GPU
 def auto_ensemble_process(
     auto_input_audio_file,
     selected_models,