Update app.py

app.py

@@ -36,7 +36,7 @@ def generate_music_tensors(description, duration: int):
 
 def save_audio(samples: torch.Tensor, filename):
     sample_rate = 30000
-    save_path = "/content/drive/MyDrive/Colab Notebooks/audio_output"
+    save_path = "audio_output"
     assert samples.dim() == 2 or samples.dim() == 3
 
     samples = samples.detach().cpu()
@@ -96,10 +96,16 @@ def main():
         music_tensors = generate_music_tensors(description, time_slider)
 
         idx = 0
-        audio_path = save_audio(music_tensors[idx], "audio_output")
-        audio_file = open(audio_path, 'rb')
+        # audio_path = save_audio(music_tensors[idx], "audio_output")
+        # audio_file = open(audio_path, 'rb')
+        # audio_bytes = audio_file.read()
+        music_tensor = music_tensors[idx]
+        save_music_file = save_audio(music_tensor)
+        audio_filepath = f'audio_output/audio_{idx}.wav'
+        audio_file = open(audio_filepath, 'rb')
         audio_bytes = audio_file.read()
 
+
         st.audio(audio_bytes, format='audio/wav')
         st.markdown(get_binary_file_downloader_html(audio_path, f'Audio_{idx}'), unsafe_allow_html=True)