thecollabagepatch commited on
Commit
3278943
·
1 Parent(s): 1b277e0

uh oh pickling error

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,7 +28,7 @@ def preprocess_audio(waveform):
28
  # processed_waveform_np = rms_normalize(peak_normalize(waveform_np))
29
  return torch.from_numpy(waveform_np).unsqueeze(0).to(device)
30
 
31
- @spaces.GPU(10)
32
  def generate_drum_sample():
33
  model = MusicGen.get_pretrained('pharoAIsanders420/micro-musicgen-jungle')
34
  model.set_generation_params(duration=10)
@@ -41,7 +41,7 @@ def generate_drum_sample():
41
 
42
  return filename_with_extension
43
 
44
- @spaces.GPU(10)
45
  def continue_drum_sample(existing_audio_path):
46
  # Load the existing audio
47
  existing_audio, sr = torchaudio.load(existing_audio_path)
 
28
  # processed_waveform_np = rms_normalize(peak_normalize(waveform_np))
29
  return torch.from_numpy(waveform_np).unsqueeze(0).to(device)
30
 
31
+ #@spaces.GPU(10)
32
  def generate_drum_sample():
33
  model = MusicGen.get_pretrained('pharoAIsanders420/micro-musicgen-jungle')
34
  model.set_generation_params(duration=10)
 
41
 
42
  return filename_with_extension
43
 
44
+ #@spaces.GPU(10)
45
  def continue_drum_sample(existing_audio_path):
46
  # Load the existing audio
47
  existing_audio, sr = torchaudio.load(existing_audio_path)