ypedz commited on
Commit
a61afaa
·
verified ·
1 Parent(s): 9f40564

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +29 -0
app.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import gradio as gr
3
+ from voxcpm import VoxCPM
4
+ import soundfile as sf
5
+ import tempfile
6
+
7
+ model = None
8
+
9
+ def load_model():
10
+ global model
11
+ if model is None:
12
+ model = VoxCPM.from_pretrained("openbmb/VoxCPM2", load_denoiser=False)
13
+ return model
14
+
15
+ def generate_speech(text):
16
+ m = load_model()
17
+ wav = m.generate(text=text, cfg_value=2.0, inference_timesteps=10)
18
+ tmp = tempfile.NamedTemporaryFile(suffix=".wav", delete=False)
19
+ sf.write(tmp.name, wav, m.tts_model.sample_rate)
20
+ return tmp.name
21
+
22
+ demo = gr.Interface(
23
+ fn=generate_speech,
24
+ inputs=gr.Textbox(label="Masukkan teks"),
25
+ outputs=gr.Audio(label="Hasil suara"),
26
+ title="VoxCPM TTS API",
27
+ )
28
+
29
+ demo.launch()