Prgckwb commited on
Commit
7a1ae19
·
verified ·
1 Parent(s): b2905f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,6 +2,7 @@ import os
2
 
3
  import gradio as gr
4
  import nltk
 
5
 
6
  os.system("python -m unidic download")
7
  from melo.api import TTS # noqa: E402
@@ -13,7 +14,7 @@ device = "auto"
13
  model = TTS(language="EN", device=device)
14
  speaker_ids = model.hps.data.spk2id
15
 
16
-
17
  def inference(
18
  text: str, speed: float, speaker: str, progress=gr.Progress(track_tqdm=True)
19
  ):
 
2
 
3
  import gradio as gr
4
  import nltk
5
+ import spaces
6
 
7
  os.system("python -m unidic download")
8
  from melo.api import TTS # noqa: E402
 
14
  model = TTS(language="EN", device=device)
15
  speaker_ids = model.hps.data.spk2id
16
 
17
+ @spaces.GPU
18
  def inference(
19
  text: str, speed: float, speaker: str, progress=gr.Progress(track_tqdm=True)
20
  ):