thecollabagepatch commited on
Commit
ee282eb
1 Parent(s): 4cd41c1

gpu durations back to 120

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -122,7 +122,7 @@ def generate_midi(seed, use_chords, chord_progression, bpm):
122
  sample_rate = 44100 # Assuming fixed sample rate from fluidsynth command
123
  return wav_filename
124
 
125
- @spaces.GPU(duration=90)
126
  def generate_music(wav_filename, prompt_duration, musicgen_model, num_iterations, bpm):
127
  # Load the audio from the passed file path
128
  song, sr = torchaudio.load(wav_filename)
@@ -184,7 +184,7 @@ def generate_music(wav_filename, prompt_duration, musicgen_model, num_iterations
184
 
185
  return combined_audio_filename
186
 
187
- @spaces.GPU(duration=90)
188
  def continue_music(input_audio_path, prompt_duration, musicgen_model, num_iterations, bpm):
189
  # Load the audio from the given file path
190
  song, sr = torchaudio.load(input_audio_path)
 
122
  sample_rate = 44100 # Assuming fixed sample rate from fluidsynth command
123
  return wav_filename
124
 
125
+ @spaces.GPU(duration=120)
126
  def generate_music(wav_filename, prompt_duration, musicgen_model, num_iterations, bpm):
127
  # Load the audio from the passed file path
128
  song, sr = torchaudio.load(wav_filename)
 
184
 
185
  return combined_audio_filename
186
 
187
+ @spaces.GPU(duration=120)
188
  def continue_music(input_audio_path, prompt_duration, musicgen_model, num_iterations, bpm):
189
  # Load the audio from the given file path
190
  song, sr = torchaudio.load(input_audio_path)