Create handler.py

handler.py

@@ -0,0 +1,19 @@
+from transformers import AutoProcessor, MusicgenForConditionalGeneration
+
+# Cargar el procesador y el modelo preentrenado
+processor = AutoProcessor.from_pretrained("facebook/musicgen-large")
+model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-large")
+
+def generate_music(prompt, guidance_scale=3, max_new_tokens=256):
+    inputs = processor(
+        text=[prompt],
+        padding=True,
+        return_tensors="pt",
+    )
+    audio_values = model.generate(
+        **inputs,
+        do_sample=True,
+        guidance_scale=guidance_scale,
+        max_new_tokens=max_new_tokens
+    )
+    return audio_values