sim04ful commited on
Commit
1a6a41a
1 Parent(s): c74f8e0

added format

Browse files
Files changed (1) hide show
  1. handler.py +3 -1
handler.py CHANGED
@@ -55,7 +55,9 @@ class EndpointHandler:
55
  enable_text_splitting=True,
56
  )
57
  audio_file = io.BytesIO()
58
- torchaudio.save(audio_file, torch.tensor(out["wav"]).unsqueeze(0), 24000)
 
 
59
  inference_time = time.time() - t0
60
  print(f"I: Time to generate audio: {inference_time} seconds")
61
  audio_str = base64.b64encode(audio_file.getvalue()).decode("utf-8")
 
55
  enable_text_splitting=True,
56
  )
57
  audio_file = io.BytesIO()
58
+ torchaudio.save(
59
+ audio_file, torch.tensor(out["wav"]).unsqueeze(0), 24000, format="wav"
60
+ )
61
  inference_time = time.time() - t0
62
  print(f"I: Time to generate audio: {inference_time} seconds")
63
  audio_str = base64.b64encode(audio_file.getvalue()).decode("utf-8")