import gradio as gr 
import torch 
from transformers import AutoProcessor 
from audiocraft.models import MusicGen
 
# Cargar el modelo MusicGen
model = MusicGen.get_pretrained('facebook/musicgen-small')
 
# Función para generar música
def generar_musica(descripcion): 
    model.set_generation_params(duration=10)  # Duración en segundos 
    output = model.generate(descriptions=[descripcion]) 
    output_path = "output.wav"
    model.save_wav(output[0], output_path) 
    return output_path
 
demo = gr.Interface( 
    fn=generar_musica,
    inputs=gr.Textbox(label="Descripción de la música"),
    outputs=gr.Audio(label="Música Generada"), 
    title="Generador de Música AI", 
    description="Introduce una descripción y genera una pista de música."
)
 
if __name__ == "__main__":
    demo.launch()