hungchiayu commited on
Commit
c75a416
1 Parent(s): 0794579

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -255,6 +255,7 @@ def gradio_generate(prompt, output_format, steps, guidance):
255
  output_wave = pipe(prompt,steps,guidance) ## Using pipeliine automatically uses flash attention for torch2.0 above
256
  #output_wave = tango.generate(prompt, steps, guidance)
257
  # output_filename = f"{prompt.replace(' ', '_')}_{steps}_{guidance}"[:250] + ".wav"
 
258
  output_filename = "temp.wav"
259
  wavio.write(output_filename, output_wave, rate=16000, sampwidth=2)
260
 
 
255
  output_wave = pipe(prompt,steps,guidance) ## Using pipeliine automatically uses flash attention for torch2.0 above
256
  #output_wave = tango.generate(prompt, steps, guidance)
257
  # output_filename = f"{prompt.replace(' ', '_')}_{steps}_{guidance}"[:250] + ".wav"
258
+ output_wave = output_wave.audios[0]
259
  output_filename = "temp.wav"
260
  wavio.write(output_filename, output_wave, rate=16000, sampwidth=2)
261