sobadatthis

app.py

@@ -121,11 +121,11 @@ def generate_midi(seed, use_chords, chord_progression, bpm):
     return (wav_filename, sample_rate)
 
 @spaces.GPU(duration=120)
-def generate_music(midi_audio_path, prompt_duration, musicgen_model, num_iterations, bpm):
-    # Load the generated audio
-    song, sr = torchaudio.load(midi_audio_path)
+def generate_music(midi_audio_info, prompt_duration, musicgen_model, num_iterations, bpm):
+    wav_filename, sample_rate = midi_audio_info
+    # Correctly use wav_filename to load the audio
+    song, sr = torchaudio.load(wav_filename)
     song = song.to(device)
-
     # Use the user-provided BPM value for duration calculation
     duration = calculate_duration(bpm)
 
@@ -242,7 +242,7 @@ with gr.Blocks() as iface:
 
 midi_result = gr.Variable()
 
-generate_midi_button.click(generate_midi, inputs=[seed, use_chords, chord_progression, bpm], outputs=midi_result)
-generate_music_button.click(generate_music, inputs=[midi_result, prompt_duration, musicgen_model, num_iterations, bpm], outputs=output_audio)
+generate_midi_button.click(generate_midi, inputs=[seed, use_chords, chord_progression, bpm], outputs=[midi_audio])
+generate_music_button.click(generate_music, inputs=[midi_audio, prompt_duration, musicgen_model, num_iterations, bpm], outputs=output_audio)
 
 iface.launch()