continuing continuation attempt 1

app.py

@@ -120,7 +120,7 @@ def generate_midi(seed, use_chords, chord_progression, bpm):
     sample_rate = 44100  # Assuming fixed sample rate from fluidsynth command
     return wav_filename
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=60)
 def generate_music(wav_filename, prompt_duration, musicgen_model, num_iterations, bpm):
     # Load the audio from the passed file path
     song, sr = torchaudio.load(wav_filename)
@@ -182,7 +182,8 @@ def generate_music(wav_filename, prompt_duration, musicgen_model, num_iterations
 
     return combined_audio_filename
 
-    def continue_music(input_audio_path, prompt_duration, musicgen_model, num_iterations, bpm):
+@spaces.GPU(duration=60)
+def continue_music(input_audio_path, prompt_duration, musicgen_model, num_iterations, bpm):
     # Load the audio from the given file path
     song, sr = torchaudio.load(input_audio_path)
     song = song.to(device)