mattricesound commited on
Commit
22b77e6
·
1 Parent(s): 517592c

Update output paths

Browse files
Files changed (1) hide show
  1. inference.py +5 -1
inference.py CHANGED
@@ -1,6 +1,7 @@
1
  from audiocraft.models import musicgen
2
  import torchaudio
3
  import soundfile as sf
 
4
 
5
  print("Loading model...")
6
  # model = musicgen.MusicGen.get_pretrained("melody")
@@ -13,4 +14,7 @@ print("Running inference...")
13
  # wav = model.generate_with_chroma(descriptions, melody[None].expand(1, -1, -1), sr)
14
  wav = model.generate(descriptions) # generates 3 samples.
15
  model_sampling_rate = 32000
16
- sf.write("output.wav", wav[0].numpy().T, model_sampling_rate)
 
 
 
 
1
  from audiocraft.models import musicgen
2
  import torchaudio
3
  import soundfile as sf
4
+ import os
5
 
6
  print("Loading model...")
7
  # model = musicgen.MusicGen.get_pretrained("melody")
 
14
  # wav = model.generate_with_chroma(descriptions, melody[None].expand(1, -1, -1), sr)
15
  wav = model.generate(descriptions) # generates 3 samples.
16
  model_sampling_rate = 32000
17
+
18
+ if not os.path.exists("data"):
19
+ os.makedirs("data")
20
+ sf.write("data/output.wav", wav[0].numpy().T, model_sampling_rate)