fix the bugs

app.py

@@ -136,7 +136,7 @@ def UI():
 							)
 					with gr.Row():
 						submit = gr.Button("Generate Music")
-						output = gr.Audio("listen to the generated music")
+						output = gr.Audio("listen to the generated music", type="filepath")
 
 			submit.click(
 				fn=predict,

gradio_components/prediction.py

@@ -103,4 +103,4 @@ def predict(model_path, text, melody, duration, topk, topp, temperature, target_
         top_p=topp,
         temperature=temperature,
         gradio_progress=progress)
-    return wavs
+    return wavs[0]