Update app.py

app.py

@@ -1,5 +1,5 @@
 import torch
-print(torch.version.cuda)
+print(f"Torch version: {torch.version.cuda}")
 
 from stable_diffusion_tf.stable_diffusion import StableDiffusion as StableDiffusionPy
 import gradio as gr
@@ -31,7 +31,7 @@ def generate_images(prompt: str, num_steps: int, unconditional_guidance_scale: i
     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)
+    final_video = gr.make_waveform("output.wav", bg_image=img[0])
     return final_video