bluenevus commited on
Commit
897a611
·
verified ·
1 Parent(s): fc26ff1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -185,6 +185,10 @@ def generate_audio(script_output, voice1, voice2, num_hosts, temperature, top_p,
185
  with torch.no_grad():
186
  paragraph_audio = snac_model(codes_tensor)
187
 
 
 
 
 
188
  paragraph_audio = paragraph_audio.cpu().numpy().flatten()
189
 
190
  silences = detect_silence(paragraph_audio)
 
185
  with torch.no_grad():
186
  paragraph_audio = snac_model(codes_tensor)
187
 
188
+ # Handle tuple output
189
+ if isinstance(paragraph_audio, tuple):
190
+ paragraph_audio = paragraph_audio[0] # Assume the first element is the audio tensor
191
+
192
  paragraph_audio = paragraph_audio.cpu().numpy().flatten()
193
 
194
  silences = detect_silence(paragraph_audio)