LTH001 commited on
Commit
6803535
·
verified ·
1 Parent(s): 009b507

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -29,9 +29,10 @@ def story_to_speech(story_text):
29
  tts_pipe = pipeline("text-to-speech", model="suno/bark-small")
30
  audio_output = tts_pipe(story_text[:400])
31
 
32
- # Convert numpy array to bytes using scipy
33
  audio_bytes = io.BytesIO()
34
- write_wav(audio_bytes, audio_output["sampling_rate"], audio_output["audio"])
 
35
  audio_bytes.seek(0)
36
 
37
  return audio_bytes
 
29
  tts_pipe = pipeline("text-to-speech", model="suno/bark-small")
30
  audio_output = tts_pipe(story_text[:400])
31
 
32
+ # Convert to bytes using numpy directly
33
  audio_bytes = io.BytesIO()
34
+ audio_np = (audio_output["audio"] * 32767).astype(np.int16)
35
+ write_wav(audio_bytes, audio_output["sampling_rate"], audio_np)
36
  audio_bytes.seek(0)
37
 
38
  return audio_bytes