Create app.py

app.py

@@ -0,0 +1,17 @@
+import gradio as gr
+import torch
+from diffusers import DiffusionPipeline
+
+# Cargar el modelo preentrenado de Hugging Face
+model = DiffusionPipeline.from_pretrained("harmonai/ddpm-256")
+model.to("cuda" if torch.cuda.is_available() else "cpu")
+
+def generate_beat(prompt):
+    audio = model(prompt, num_inference_steps=50)
+    audio_path = "generated_beat.wav"
+    audio.save(audio_path)
+    return audio_path
+
+# Crear interfaz de usuario con Gradio
+iface = gr.Interface(fn=generate_beat, inputs="text", outputs="audio")
+iface.launch()