Update app.py

app.py

@@ -52,7 +52,7 @@ def style_transfer(prompt, negative_prompt, audio_input):
     wav = wav_bytes_from_spectrogram_image(new_spectro[0])
     with open("output.wav", "wb") as f:
         f.write(wav[0].getbuffer())
-    return new_spectro, 'output.wav', gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+    return new_spectro[0], 'output.wav', gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 def image_from_spectrogram(
     spectrogram: np.ndarray, max_volume: float = 50, power_for_image: float = 0.25