Update app.py

app.py

@@ -15,17 +15,16 @@ def generate_music(description):
     # Generate music based on the provided description
     wav = model.generate([description])
     
-    # # Save the generated audio as "output.wav" with loudness normalization at -14 dB LUFS.
-    # audio_write("output.wav", wav[0].cpu(), model.sample_rate, strategy="loudness")
+    # Convert the audio waveform to a format that Gradio can display
+    audio_data = wav[0].cpu().numpy().tobytes()
 
-    # Return the generated audio for playback
-    return wav[0].cpu().numpy()
+    return audio_data  # Return the audio data
 
 # Define a Gradio interface
 iface = gr.Interface(
     fn=generate_music,  # Use the generate_music function for processing input
     inputs="text",  # Accept text input from the user
-    outputs="audio",  # Specify "audio" to indicate streaming audio output
+    outputs=gr.Audio(type="numpy", label="Audio"),  # Display the generated audio as output
     title="Music Generation from Descriptions",  # Set the title of the web interface
     description="Generate music based on descriptions.",  # Provide a description
     live=False  # Set to False if you don't want real-time updates (for beginner-friendly interaction)