Update app.py

app.py

@@ -1,18 +1,27 @@
-import gradio as gr
+ import gradio as gr
 import torch
-from transformers import AutoProcessor
+from transformers import AutoProcessor 
 from audiocraft.models import MusicGen
 
-def probar_imports():
-    return "¡Las importaciones funcionan sin errores!"
+#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=probar_imports,
-    inputs=None,
-    outputs="text",
-    title="Prueba de imports",
-    description="Si esto funciona, significa que no hay error de sintaxis."
+    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()
+