Update app.py

app.py

@@ -356,16 +356,19 @@ def generate_music(
             cutoff_freq=5500.0
         )
         print("All process Done")
-        return final_output_path
+        
+        # Load the final audio file and return the numpy array
+        final_audio, sr = torchaudio.load(final_output_path)
+        return (sr, final_audio.squeeze().numpy())
 
 
 @spaces.GPU(duration=120)
 def infer(genre_txt_content, lyrics_txt_content, num_segments=2, max_new_tokens=10):
     # Execute the command
     try:
-        music = generate_music(genre_txt=genre_txt_content, lyrics_txt=lyrics_txt_content, run_n_segments=num_segments,
+        audio_data = generate_music(genre_txt=genre_txt_content, lyrics_txt=lyrics_txt_content, run_n_segments=num_segments,
                                cuda_idx=0, max_new_tokens=max_new_tokens)
-        return music
+        return audio_data
     except Exception as e:
         gr.Warning("An Error Occured: " + str(e))
         return None