Update app.py (#9)

- Update app.py (04850d2fd81da8bcc056d14ab98908b27513582c)


Co-authored-by: oat <[email protected]>

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)