SAUL19 commited on
Commit
ecba411
·
1 Parent(s): dcfa0af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -96,7 +96,8 @@ def generateAudio(text_to_audio, s3_save_as):
96
  speaker_embeddings = torch.randn((1, 512)).to(device)
97
  # generate speech with the models
98
  speech = model.generate_speech(
99
- inputs["input_ids"], speaker_embeddings, vocoder=vocoder)
 
100
 
101
  # create BytesIO object to store the audio
102
  audio_buffer = BytesIO()
 
96
  speaker_embeddings = torch.randn((1, 512)).to(device)
97
  # generate speech with the models
98
  speech = model.generate_speech(
99
+ inputs["input_ids"], speaker_embeddings, vocoder=vocoder, max_length=5000)
100
+
101
 
102
  # create BytesIO object to store the audio
103
  audio_buffer = BytesIO()