Update app.py

app.py

@@ -25,7 +25,7 @@ def generate_images(prompt: str, num_steps: int, unconditional_guidance_scale: i
     )
 
     pil_img = Image.fromarray(img[0])
-    wav = wav_bytes_from_spectrogram_image(new_spectro[0])
+    wav = wav_bytes_from_spectrogram_image(pil_img)
     with open("output.wav", "wb") as f:
         f.write(wav[0].getbuffer())
     final_video = gr.make_waveform("output.wav", bg_image=pil_img)