Update app.py

app.py

@@ -1,27 +1,14 @@
- import gradio as gr
-import torch
-from transformers import AutoProcessor 
-from audiocraft.models import MusicGen
+ import gradio as gr import torch from transformers import AutoProcessor from audiocraft.models import MusicGen
+
+Cargar el modelo 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
+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."
-)
+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()
+if name == "main": demo.launch()