Spaces:
Runtime error
Runtime error
File size: 1,049 Bytes
65f6ccd 1ad1d98 65f6ccd 1ad1d98 65f6ccd b8e52fb 65f6ccd 741419c 1ad1d98 741419c 1ad1d98 08d8fc7 1ad1d98 e359ea8 1ad1d98 65f6ccd 1ad1d98 65f6ccd 1ad1d98 53d6104 1ad1d98 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import gradio as gr
import torch
import torchaudio
from diffusers import AudioLDMPipeline
# Cargar el modelo de AudioLDM
pipe = AudioLDMPipeline.from_pretrained("cvssp/audioldm-s-full-v2")
pipe.to("cpu") # Ejecutar en CPU
# Funci贸n para generar m煤sica
def generar_musica(descripcion):
try:
audio = pipe(descripcion, num_inference_steps=25) # Reducir pasos para mejor rendimiento
audio_path = "musica_generada.wav"
torchaudio.save(audio_path, audio["audio"].unsqueeze(0).cpu(), 16000)
return audio_path
except Exception as e:
return f"Error: {e}"
# Interfaz Gradio
with gr.Blocks() as interfaz:
gr.Markdown("# Generador de M煤sica AI")
gr.Markdown("Introduce una descripci贸n y genera una pista de m煤sica.")
descripcion = gr.Textbox(label="Descripci贸n de la m煤sica")
salida = gr.Audio(label="M煤sica Generada")
boton = gr.Button("Generar M煤sica")
boton.click(fn=generar_musica, inputs=descripcion, outputs=salida)
if __name__ == "__main__":
interfaz.launch()
|