naufalhm commited on
Commit
52bf57d
·
verified ·
1 Parent(s): b1ecec8

Update backend.py

Browse files
Files changed (1) hide show
  1. backend.py +1 -4
backend.py CHANGED
@@ -30,8 +30,5 @@ def text_to_speech(text):
30
  inputs = tokenizer_tts(text, return_tensors="pt").to(model_tts.device)
31
  with torch.no_grad():
32
  audio = model_tts(**inputs).waveform.squeeze().cpu().numpy()
33
- buffer = io.BytesIO()
34
  sampling_rate = model_tts.config.sampling_rate
35
- sf.write(buffer, audio, sampling_rate, format="WAV")
36
- buffer.seek(0)
37
- return sampling_rate, buffer
 
30
  inputs = tokenizer_tts(text, return_tensors="pt").to(model_tts.device)
31
  with torch.no_grad():
32
  audio = model_tts(**inputs).waveform.squeeze().cpu().numpy()
 
33
  sampling_rate = model_tts.config.sampling_rate
34
+ return sampling_rate, audio