Update app.py

app.py

@@ -36,8 +36,8 @@ def classic(prompt, negative_prompt, duration):
 
 def style_transfer(prompt, negative_prompt, audio_input):
     spec = spectro_from_wav(audio_input)
-    new_spectro = pipe2(prompt=prompt, image=spec, strength=0.5, guidance_scale=7).images[0]
-    wav = wav_bytes_from_spectrogram_image(spec)
+    new_spectro = pipe2(prompt=prompt, image=spec, strength=0.5, guidance_scale=7).images
+    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)