spaces

app.py

@@ -12,9 +12,10 @@ Is there a long queue on this space? Duplicate it and add a more powerful GPU to
 
 **NOTE: StyleTTS 2 is _currently_ English-only. Join the Discord for updates on multilingual training.**
 """
+import spaces
 import gradio as gr
 import styletts2importable
-import ljspeechimportable
+#import ljspeechimportable
 import torch
 import os
 from txtsplit import txtsplit
@@ -44,6 +45,8 @@ for v in voicelist:
 #     # return (24000, styletts2importable.inference(text, voices[v], alpha=0.3, beta=0.7, diffusion_steps=7, embedding_scale=1))
 #     return (24000, styletts2importable.inference(text, voices[v], alpha=0.3, beta=0.7, diffusion_steps=multispeakersteps, embedding_scale=1))
 if not torch.cuda.is_available(): INTROTXT += "\n\n### You are on a CPU-only system, inference will be much slower.\n\nYou can use the [online demo](https://huggingface.co/spaces/styletts2/styletts2) for fast inference."
+
+@spaces.GPU
 def synthesize(text, voice, lngsteps, password, progress=gr.Progress()):
     if text.strip() == "":
         raise gr.Error("You must enter some text")
@@ -75,6 +78,8 @@ def synthesize(text, voice, lngsteps, password, progress=gr.Progress()):
 #         return (24000, np.concatenate(audios))
 #     else:
 #         raise gr.Error('Wrong access code')
+
+@spaces.GPU
 def clsynthesize(text, voice, vcsteps, embscale, alpha, beta, progress=gr.Progress()):
     # if text.strip() == "":
     #     raise gr.Error("You must enter some text")
@@ -101,6 +106,8 @@ def clsynthesize(text, voice, vcsteps, embscale, alpha, beta, progress=gr.Progre
         audios.append(styletts2importable.inference(t, vs, alpha=alpha, beta=beta, diffusion_steps=vcsteps, embedding_scale=embscale))
         # audios.append(styletts2importable.inference(t, vs, diffusion_steps=10, alpha=0.3, beta=0.7, embedding_scale=5))
     return (24000, np.concatenate(audios))
+
+@spaces.GPU
 def ljsynthesize(text, steps, progress=gr.Progress()):
     # if text.strip() == "":
     #     raise gr.Error("You must enter some text")