Update app.py

app.py

@@ -11,6 +11,18 @@ import gradio as gr
 
 import config
 
+try:
+    import spaces
+    USING_SPACES = True
+except ImportError:
+    USING_SPACES = False
+
+def gpu_decorator(func):
+    if USING_SPACES:
+        return spaces.GPU(func)
+    else:
+        return func
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 def ask_tos_patch(self, output_path):
@@ -25,7 +37,7 @@ for id, model in config.models.items():
     tts.download_model_by_name(model)
     models[id] = TTS.api.TTS(model).to(device)
 
-
+@gpu_decorator
 def synthesize_tts(
     text: str = 'Hello, World!',
     speaker_wavs: List[gr.File] = None,