Update app.py

app.py

@@ -35,7 +35,7 @@ def classic(prompt, negative_prompt, duration):
     wav = wav_bytes_from_spectrogram_image(spec)
     with open("output.wav", "wb") as f:
         f.write(wav[0].getbuffer())
-    return spec, 'output.wav', gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+    return spec, 'output.wav'
 
 def style_transfer(prompt, negative_prompt, audio_input):
     spec = spectro_from_wav(audio_input)
@@ -52,7 +52,7 @@ def style_transfer(prompt, negative_prompt, audio_input):
     wav = wav_bytes_from_spectrogram_image(new_spectro[0])
     with open("output.wav", "wb") as f:
         f.write(wav[0].getbuffer())
-    return new_spectro[0], 'output.wav', gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+    return new_spectro[0], 'output.wav'
 
 def image_from_spectrogram(
     spectrogram: np.ndarray, max_volume: float = 50, power_for_image: float = 0.25
@@ -228,7 +228,7 @@ with gr.Blocks(css="style.css") as demo:
         
         gr.HTML(article)
     
-    send_btn.click(predict, inputs=[prompt_input, negative_prompt, audio_input, duration_input], outputs=[spectrogram_output, sound_output, share_button, community_icon, loading_icon])
+    send_btn.click(predict, inputs=[prompt_input, negative_prompt, audio_input, duration_input], outputs=[spectrogram_output, sound_output])
     #share_button.click(None, [], [], _js=share_js)
 
 demo.queue(max_size=250).launch(debug=True)