Updated app

app.py

@@ -59,14 +59,10 @@ if st.button("Generate Music"):
         with st.spinner("Generating audio..."):
             results = generate_audio(processor, model, prompt)
         if results is not None:
-            with st.spinner("Saving audio..."):
-                filename = "plant_orchestra" + ".wav"
-                save_file(model, results, filename)
-            with st.spinner("Displaying audio..."):
-                with open(filename, "rb") as f:
-                    generation = f.read()
-                    st.write("Listen to the generated music:")
-                st.audio(generation)
+            sampling_rate = model.config.audio_encoder.sampling_rate
+            st.write("Listen to the generated music:")
+            st.audio(sample_rate=sampling_rate, data=results[0, 0].cpu().numpy(), format="audio/wav")
+            
             
 # Add additional information and instructions for users
 st.sidebar.subheader("How to Use:")