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()