Update webui.py

webui.py

@@ -6,6 +6,7 @@ import sys
 import traceback
 from pathlib import Path
 from typing import Literal, Tuple
+import spaces
 
 import numpy as np
 import torch
@@ -169,6 +170,7 @@ class AppState:
         )
         self.phoneset_path = "soulxsinger/utils/phoneme/phone_set.json"
 
+    @spaces.GPU
     def run_preprocess(
         self,
         prompt_path: Path,
@@ -198,6 +200,7 @@ class AppState:
         except Exception as e:
             return False, f"preprocess failed: {e}"
 
+    @spaces.GPU
     def run_svs(
         self,
         control: str,
@@ -234,6 +237,7 @@ class AppState:
         except Exception as e:
             return False, f"svs inference failed: {e}", None, prompt_meta_path, target_meta_path
 
+    @spaces.GPU
     def run_svs_from_paths(
         self,
         prompt_wav_path: str,
@@ -364,7 +368,7 @@ def _resolve_file_path(x):
         x = x[0]
     return x if (x and os.path.isfile(x)) else None
 
-
+@spaces.GPU
 def transcription_function(
     prompt_audio,
     target_audio,
@@ -431,7 +435,7 @@ def transcription_function(
         print(traceback.format_exc(), file=sys.stderr, flush=True)
         return None, None
 
-
+@spaces.GPU
 def synthesis_function(
     prompt_audio,
     prompt_metadata,