Update app.py

app.py

@@ -1,16 +1,14 @@
- import gradio as gr 
+import gradio as gr 
 import torch 
-from transformers import AutoProcessor
+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
+    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) 
@@ -21,7 +19,8 @@ demo = gr.Interface(
     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." )
+    description="Introduce una descripción y genera una pista de música."
+)
 
 if name == "main":
     demo.launch()