add infer logic

app.py

@@ -64,10 +64,9 @@ def generate_song(description, lyric, prompt_audio=None):
     if prompt_audio is not None:
         print("Using prompt audio for generation")
     
-    # 从文件中加载示例音频
-    audio_path = os.path.join(os.path.dirname(__file__), "sample/example.mp3")
-    audio_data, sample_rate = librosa.load(audio_path, sr=None)  # 保持原始采样率
+    sample_rate = model.cfg.sample_rate
     
+    audio_data = model(description, lyric, prompt_audio).cpu().float().numpy()
     
     # 创建输入配置的JSON
     input_config = {
@@ -75,7 +74,6 @@ def generate_song(description, lyric, prompt_audio=None):
         "lyric": lyric,
         "has_prompt_audio": prompt_audio is not None,
         "timestamp": datetime.now().isoformat(),
-        "model": str(model),
     }
     
     return (sample_rate, audio_data), json.dumps(input_config, indent=2)