Upload 2 files

handler.py

@@ -0,0 +1,20 @@
+from audiocraft.models import MusicGen
+from scipy.io.wavfile import write
+import torch
+import io
+
+class EndpointHandler:
+    def __init__(self, path=""):
+        self.model = MusicGen.get_pretrained("facebook/musicgen-large")
+        self.model.set_generation_params(duration=8)  # seconds
+
+    def __call__(self, data):
+        prompt = data.get("inputs", "calm ambient piano")
+        wav = self.model.generate([prompt])
+        sample_rate = 32000
+
+        buffer = io.BytesIO()
+        write(buffer, sample_rate, wav[0].cpu().numpy())
+        buffer.seek(0)
+
+        return {"audio": buffer.read()}

requirements.txt

@@ -0,0 +1,3 @@
+audiocraft
+torch
+scipy