Update app.py

app.py

@@ -35,7 +35,8 @@ def generate_music_tensors(descriptions, duration: int):
 
 def save_audio(samples: torch.Tensor):
     sample_rate = 30000
-    save_path = "/content/drive/MyDrive/Colab Notebooks/audio_output"
+    # 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()
@@ -97,13 +98,14 @@ def main():
             idx = 0
             music_tensor = music_tensors[idx]
             save_music_file = save_audio(music_tensor)
-            audio_filepath = f'/content/drive/MyDrive/Colab Notebooks/audio_output/audio_{idx}.wav'
+            # audio_filepath = f'/content/drive/MyDrive/Colab Notebooks/audio_output/audio_{idx}.wav'
+            audio_filepath = 'audio_output/audio_0.wav'
             audio_file = open(audio_filepath, 'rb')
             audio_bytes = audio_file.read()
 
             # Play the full audio
             st.audio(audio_bytes, format='audio/wav')
-            st.markdown(get_binary_file_downloader_html(audio_filepath, f'Audio_{idx}'), unsafe_allow_html=True)
+            st.markdown(get_binary_file_downloader_html(audio_filepath, 'Audio'), unsafe_allow_html=True)
 
 
 if __name__ == "__main__":