KGNINJA commited on
Commit
8167ecb
·
verified ·
1 Parent(s): 9580fd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -11
app.py CHANGED
@@ -6,20 +6,16 @@ from transformers import AutoProcessor, AutoModel
6
  import gradio as gr
7
 
8
 
9
- # ==========================
10
- # 1) TTSモデル(kokoro)
11
- # ==========================
12
- processor = AutoProcessor.from_pretrained("kokoro-speech/kokoro-tts")
13
- model = AutoModel.from_pretrained("kokoro-speech/kokoro-tts")
14
 
 
 
15
 
16
  def tts_standard(text):
17
- """標準イントネーションの音声生成"""
18
- inputs = processor(text=text, return_tensors="pt")
19
- with torch.no_grad():
20
- output = model(**inputs).waveform
21
- audio = output.squeeze().cpu().numpy()
22
- return 16000, audio
23
 
24
 
25
  # ==========================
 
6
  import gradio as gr
7
 
8
 
9
+ from transformers import pipeline
 
 
 
 
10
 
11
+ # 軽量TTS
12
+ tts = pipeline("text-to-speech", model="p0p0pp/KoeTTS")
13
 
14
  def tts_standard(text):
15
+ result = tts(text)
16
+ audio = result["audio"]
17
+ return (22050, audio)
18
+
 
 
19
 
20
 
21
  # ==========================