Googlefan
commited on
Commit
•
21ae21d
1
Parent(s):
7f75fe9
feat: add app.py and requirements.txt for text-to-speech functionality
Browse files- .gitignore +2 -0
- app.py +46 -0
- requirements.txt +2 -0
.gitignore
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
.aider*
|
2 |
+
venv/
|
app.py
ADDED
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import gradio as gr
|
2 |
+
from huggingface_hub import hf_hub_download
|
3 |
+
from sbv2_bindings import TTSModel
|
4 |
+
from uuid import uuid4
|
5 |
+
|
6 |
+
bert = hf_hub_download("googlefan/sbv2_onnx_models", "deberta.onnx")
|
7 |
+
tokenizer = hf_hub_download("googlefan/sbv2_onnx_models", "tokenizer.json")
|
8 |
+
model = TTSModel.from_path(bert, tokenizer)
|
9 |
+
|
10 |
+
|
11 |
+
def load_and_synthesize(text: str, path: str, sdp: float = 0.0, speed: float = 1.0):
|
12 |
+
uid = str(uuid4())
|
13 |
+
path = path.split("/")
|
14 |
+
filename = path[-1]
|
15 |
+
repo_id = "/".join(path[:-1])
|
16 |
+
model.load_sbv2file_from_path(uid, hf_hub_download(repo_id, filename))
|
17 |
+
print("All setup is done!")
|
18 |
+
style_vector = model.get_style_vector(uid, 0, 1.0)
|
19 |
+
return model.synthesize(text, uid, style_vector, sdp, 1.0 / speed)
|
20 |
+
|
21 |
+
|
22 |
+
iface = gr.Interface(
|
23 |
+
fn=load_and_synthesize, # The function to call
|
24 |
+
inputs=[
|
25 |
+
gr.Textbox(lines=2, placeholder="テキスト"), # Text input
|
26 |
+
gr.Textbox(lines=1, max_lines=1, placeholder="パス"), # Path input
|
27 |
+
gr.Slider(
|
28 |
+
minimum=0.0, maximum=1.0, step=0.01, default=0.0, label="SDP"
|
29 |
+
), # SDP input
|
30 |
+
gr.Slider(
|
31 |
+
minimum=0.5, maximum=2.0, step=0.1, default=1.0, label="Speed"
|
32 |
+
), # Speed input
|
33 |
+
],
|
34 |
+
outputs="audio",
|
35 |
+
title="SBV2音声合成",
|
36 |
+
description="テキストとモデルのパスを入力して音声を生成します。SDPと速度を調整して音声の質を変更できます。",
|
37 |
+
examples=[
|
38 |
+
"おはようございます。",
|
39 |
+
"googlefan/sbv2_personal_models/tenkas.sbv2",
|
40 |
+
0.0,
|
41 |
+
1.0,
|
42 |
+
],
|
43 |
+
)
|
44 |
+
|
45 |
+
# Launch the Gradio app
|
46 |
+
iface.launch()
|
requirements.txt
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
git+https://github.com/tuna2134/sbv2-api#egg=sbv2_bindings&subdirectory=sbv2_bindings
|
2 |
+
gradio
|