Prgckwb
commited on
Commit
•
02c46e7
1
Parent(s):
a5eaf4f
fix error
Browse files
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 |
-
|
22 |
-
model.tts_to_file(text, speaker_ids[speaker],
|
23 |
-
return
|
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__":
|