Add file download option

app.py

@@ -4,6 +4,7 @@ from PIL import Image
 from diffusers import StableDiffusionPipeline
 import gradio as gr
 import torch
+from spectro import wav_bytes_from_spectrogram_image
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16 if device == "cuda" else torch.float32
@@ -69,11 +70,10 @@ def infer(prompt, style):
     style_prompt = prompt + style
     image = get_bg_image(style_prompt)
     audio = get_music(prompt)
-    return (
-        gr.make_waveform(audio,
-                         bg_image=image,
-                         bars_color=random.choice(COLORS))
-    )
+    video = gr.make_waveform(audio,
+                             bg_image=image,
+                             bars_color=random.choice(COLORS))
+    return video, video
 
 css = """
     #col-container {max-width: 700px; margin-left: auto; margin-right: auto;}
@@ -119,7 +119,7 @@ with gr.Blocks(css=css) as demo:
         send_btn = gr.Button(value="Send", elem_id="submit-btn")   
       send_btn.click(infer,
                      inputs=[prompt_input, style_input],
-                     outputs=[gr.Video()])
+                     outputs=[gr.Video(), gr.File()])
 
       gr.Markdown("""
             [![Twitter Follow](https://img.shields.io/twitter/follow/juancopi81?style=social)](https://twitter.com/juancopi81)