Spaces:
Running
on
Zero
Running
on
Zero
hungchiayu
commited on
Update app.py
Browse files
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 |
|