Spaces:
Build error
Build error
Commit
·
22b77e6
1
Parent(s):
517592c
Update output paths
Browse files- 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 |
-
|
|
|
|
|
|
|
|
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)
|