Prgckwb commited on
Commit
02c46e7
1 Parent(s): a5eaf4f
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -4,10 +4,10 @@ import os
4
  import gradio as gr
5
  import nltk
6
 
7
- os.system("python -m unidic download")
8
  from melo.api import TTS # noqa: E402
9
 
10
- nltk.download("averaged_perceptron_tagger_eng")
11
 
12
  # Get device
13
  device = "auto"
@@ -18,9 +18,9 @@ speaker_ids = model.hps.data.spk2id
18
  def inference(
19
  text: str, speed: float, speaker: str, progress=gr.Progress(track_tqdm=True)
20
  ):
21
- bio = io.BytesIO()
22
- model.tts_to_file(text, speaker_ids[speaker], bio, speed=speed)
23
- return bio
24
 
25
 
26
  if __name__ == "__main__":
 
4
  import gradio as gr
5
  import nltk
6
 
7
+ # os.system("python -m unidic download")
8
  from melo.api import TTS # noqa: E402
9
 
10
+ # nltk.download("averaged_perceptron_tagger_eng")
11
 
12
  # Get device
13
  device = "auto"
 
18
  def inference(
19
  text: str, speed: float, speaker: str, progress=gr.Progress(track_tqdm=True)
20
  ):
21
+ out_path = "audio.wav"
22
+ model.tts_to_file(text, speaker_ids[speaker], out_path, speed=speed, format='wav')
23
+ return out_path
24
 
25
 
26
  if __name__ == "__main__":