Update app.py

app.py

@@ -21,13 +21,16 @@ def generar_musica(descripcion):
         # Generar música
         output = model.generate(descriptions=[descripcion])
 
-        if not output:
-            print("Error: No se generó ninguna salida.")
+        if not isinstance(output, torch.Tensor):
+            print("Error: La salida del modelo no es un tensor válido.")
             return "Error: No se pudo generar la música."
 
+        # Convertir el tensor a numpy
+        audio_data = output[0].cpu().detach().numpy()
+
         # Guardar el archivo de audio correctamente
         output_path = f"output_{uuid.uuid4().hex}.wav"
-        torchaudio.save(output_path, output[0].unsqueeze(0), 32000)  # Guardar audio con sample rate 32kHz
+        torchaudio.save(output_path, torch.tensor(audio_data).unsqueeze(0), 32000)  # Sample rate 32kHz
 
         print(f"Música generada y guardada en {output_path}")
         return output_path