ninumm commited on
Commit
66b9e70
·
1 Parent(s): 455d935

change to audio

Browse files
Files changed (1) hide show
  1. ttsFB.py +4 -2
ttsFB.py CHANGED
@@ -18,9 +18,11 @@ def tts(lang, text):
18
  with torch.no_grad():
19
  output = model(**inputs).waveform
20
 
21
- output_np = output.squeeze().cpu().numpy() # 去除多餘維度並轉為 NumPy
 
 
22
  # scipy.io.wavfile.write(f"{fileName}.wav", rate=model.config.sampling_rate, data=output_np)
23
- return output_np
24
 
25
  iface = gr.Interface(
26
  fn=tts,
 
18
  with torch.no_grad():
19
  output = model(**inputs).waveform
20
 
21
+ audio = gr.Audio(type="numpy", label="語音輸出") # 使用 numpy 格式
22
+
23
+ # output_np = output.squeeze().cpu().numpy() # 去除多餘維度並轉為 NumPy
24
  # scipy.io.wavfile.write(f"{fileName}.wav", rate=model.config.sampling_rate, data=output_np)
25
+ return audio
26
 
27
  iface = gr.Interface(
28
  fn=tts,