Update app.py

app.py

@@ -1,14 +1,28 @@
- 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
+#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
+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()