Update app.py

app.py

@@ -35,7 +35,8 @@ def classic(prompt, negative_prompt, duration):
     # spec = pipe(prompt, negative_prompt=negative_prompt, height=512, width=512).images[0]
     spec = pipe2(prompt=prompt, negative_prompt=negative_prompt, image=im, strength=0.5, guidance_scale=7).images
     print(spec)
-    wav = wav_bytes_from_spectrogram_image(spec)
+    # wav = wav_bytes_from_spectrogram_image(spec)
+    wav = wav_bytes_from_spectrogram_image(spec[0])
     with open("output.wav", "wb") as f:
         f.write(wav[0].getbuffer())
     return spec, 'output.wav', gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)