Update app.py

app.py

@@ -28,10 +28,11 @@ def generate_images(prompt: str, num_steps: int, unconditional_guidance_scale: i
     )
 
     pil_img = Image.fromarray(img[0])
+    pil_img.save("img.png")
     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=img[0])
+    final_video = gr.make_waveform("output.wav", bg_image="img.png")
     return final_video