fix dimension bug

app.py

@@ -66,7 +66,7 @@ def generate_song(description, lyric, prompt_audio=None):
     
     sample_rate = model.cfg.sample_rate
     
-    audio_data = model(lyric, description, op.join(APP_DIR, 'sample/19_2-又是一天过去，烦恼如影随形10s.wav')).cpu().float().numpy()
+    audio_data = model(lyric, description, op.join(APP_DIR, 'sample/19_2-又是一天过去，烦恼如影随形10s.wav')).cpu().permute(1, 0).float().numpy()
     
     # 创建输入配置的JSON
     input_config = {
@@ -78,7 +78,6 @@ def generate_song(description, lyric, prompt_audio=None):
     
     return (sample_rate, audio_data), json.dumps(input_config, indent=2)
 
-
 # 创建Gradio界面
 with gr.Blocks(title="LeVo Demo Space") as demo:
     gr.Markdown("# 🎵 LeVo Demo Space")