Update app.py

app.py

@@ -2,6 +2,7 @@
 import os
 import logging
 import re_matching
+import spaces
 from tools.sentence import split_by_language
 
 logging.getLogger("numba").setLevel(logging.WARNING)
@@ -35,7 +36,7 @@ device = config.webui_config.device
 if device == "mps":
     os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
 
-
+@spaces.GPU
 def generate_audio(
     slices,
     sdp_ratio,
@@ -79,7 +80,7 @@ def generate_audio(
             audio_list.append(audio16bit)
     return audio_list
 
-
+@spaces.GPU
 def generate_audio_multilang(
     slices,
     sdp_ratio,
@@ -119,7 +120,7 @@ def generate_audio_multilang(
             audio_list.append(audio16bit)
     return audio_list
 
-
+@spaces.GPU
 def tts_split(
     text: str,
     speaker,
@@ -191,7 +192,7 @@ def tts_split(
     audio_concat = np.concatenate(audio_list)
     return ("Success", (hps.data.sampling_rate, audio_concat))
 
-
+@spaces.GPU
 def process_mix(slice):
     _speaker = slice.pop()
     _text, _lang = [], []
@@ -211,7 +212,7 @@ def process_mix(slice):
                 _lang += [[lang] for part in content[1:]]
     return _text, _lang, _speaker
 
-
+@spaces.GPU
 def process_auto(text):
     _text, _lang = [], []
     for slice in text.split("|"):
@@ -228,7 +229,7 @@ def process_auto(text):
         _lang.append(temp_lang)
     return _text, _lang
 
-
+@spaces.GPU
 def process_text(
     text: str,
     speaker,
@@ -303,7 +304,7 @@ def process_text(
         )
     return audio_list
 
-
+@spaces.GPU
 def tts_fn(
     text: str,
     speaker,
@@ -345,7 +346,7 @@ def tts_fn(
     audio_concat = np.concatenate(audio_list)
     return "Success", (hps.data.sampling_rate, audio_concat)
 
-
+@spaces.GPU
 def format_utils(text, speaker):
     _text, _lang = process_auto(text)
     res = f"[{speaker}]"
@@ -355,13 +356,13 @@ def format_utils(text, speaker):
         res += "|"
     return "mix", res[:-1]
 
-
+@spaces.GPU
 def load_audio(path):
     audio, sr = librosa.load(path, 48000)
     # audio = librosa.resample(audio, 44100, 48000)
     return sr, audio
 
-
+@spaces.GPU
 def gr_util(item):
     if item == "Text prompt":
         return {"visible": True, "__type__": "update"}, {