No write

app.py

@@ -1,8 +1,6 @@
 import gradio as gr
 import json
 import torch
-import wavio
-import spaces
 
 from tqdm import tqdm
 from huggingface_hub import snapshot_download
@@ -81,15 +79,10 @@ tango = Tango(device = "cpu")
 tango.vae.to(device_type)
 tango.stft.to(device_type)
 tango.model.to(device_type)
-    
-@spaces.GPU(duration = 60)
+
 def gradio_generate(prompt, steps, guidance):
     output_wave = tango.generate(prompt, steps, guidance)
-    # output_filename = f"{prompt.replace(' ', '_')}_{steps}_{guidance}"[:250] + ".wav"
-    output_filename = "temp.wav"
-    wavio.write(output_filename, output_wave, rate = 16000, sampwidth = 2)
-    
-    return output_filename
+    return gr.make_waveform((16000, output_wave))
 
 description_text = """
         <p style="text-align: center;">
@@ -111,10 +104,11 @@ description_text = """
         """
 # Gradio input and output components
 input_text = gr.Textbox(label = "Prompt", value = "Snort of a horse", lines = 2, autofocus = True)
-output_audio = gr.Audio(label = "Generated Audio", type = "filepath")
 denoising_steps = gr.Slider(label = "Steps", minimum = 100, maximum = 200, value = 100, step = 1, interactive = True)
 guidance_scale = gr.Slider(label = "Guidance Scale", minimum = 1, maximum = 10, value = 3, step = 0.1, interactive = True)
 
+output_audio = gr.Audio(label = "Generated Audio")
+
 # Gradio interface
 gr_interface = gr.Interface(
     fn = gradio_generate,