Update generate_beat.py

generate_beat.py

@@ -1,14 +1,15 @@
 import torch
 from diffusers import DiffusionPipeline
 
-# Cargar modelo preentrenado desde Hugging Face
-model = DiffusionPipeline.from_pretrained("harmonai/ddpm-256")
+# Cargar un modelo preentrenado público desde Hugging Face
+model = DiffusionPipeline.from_pretrained("CompVis/ldm-text2im-large-256")
 model.to("cuda" if torch.cuda.is_available() else "cpu")
 
-# Generar beat
+# Generar beat con el prompt deseado
 prompt = "trap beat with deep bass and snare rolls"
 audio = model(prompt, num_inference_steps=50)
 
 # Guardar el beat generado
 audio.save("trap_beat.wav")
 print("Beat generado: trap_beat.wav")
+