Update app.py

app.py

@@ -3,6 +3,8 @@ import gradio as gr
 from tensorflow import keras
 from PIL import Image
 
+from spectro import wav_bytes_from_spectrogram_image
+
 keras.mixed_precision.set_global_policy("float32")
 # load keras model
 resolution=512
@@ -23,8 +25,11 @@ def generate_images(prompt: str, num_steps: int, unconditional_guidance_scale: i
     )
 
     pil_img = Image.fromarray(img[0])
-
-    return pil_img
+    wav = wav_bytes_from_spectrogram_image(new_spectro[0])
+    with open("output.wav", "wb") as f:
+        f.write(wav[0].getbuffer())
+    final_video = gr.make_waveform("output.wav", bg_image=pil_img)
+    return final_video
 
 
 # pass function, input type for prompt, the output for multiple images
@@ -41,7 +46,7 @@ gr.Interface(
         gr.Slider(label='Temperature', value=1, maximum=1.5, minimum=0, step=0.1),
     ], 
     outputs=[
-        gr.Image(),
+        gr.Video(),
     ],
     examples=[["a $currulao song", 50, 7.5, 1]],
     ).queue().launch(debug=True)