added tempfile

app.py

@@ -10,6 +10,7 @@ from audiocraft.models import MusicGen
 from audiocraft.data.audio import audio_write
 from pydub import AudioSegment
 import spaces
+import tempfile
 
 # Check if CUDA is available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -120,10 +121,12 @@ def generate_midi(seed, use_chords, chord_progression, bpm):
 
 @spaces.GPU(duration=120)
 def generate_music(midi_audio, prompt_duration, musicgen_model, num_iterations, bpm):
-    wav_filename = midi_audio[0]  # Extract the file path from the tuple
+    with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as temp_file:
+        temp_filename = temp_file.name
+        torchaudio.save(temp_filename, midi_audio, sample_rate=44100)
 
     # Load the generated audio
-    song, sr = torchaudio.load(wav_filename)
+    song, sr = torchaudio.load(temp_filename)
     song = song.to(device)
 
     # Use the user-provided BPM value for duration calculation
@@ -178,7 +181,7 @@ def generate_music(midi_audio, prompt_duration, musicgen_model, num_iterations,
     combined_audio.export(combined_audio_filename, format="mp3")
 
     # Clean up temporary files
-    os.remove(wav_filename)
+    os.remove(temp_filename)
     for filename in all_audio_files:
         os.remove(filename)