Update app.py

app.py

@@ -36,7 +36,11 @@ def generate_music_tensors(descriptions, duration: int):
 
 def save_audio(samples: torch.Tensor):
     sample_rate = 30000
-    save_path = "/audio_output"
+    save_path = "/tmp/audio_output"  # Use /tmp directory
+
+    if not os.path.exists(save_path):
+        os.makedirs(save_path)
+
     assert samples.dim() == 2 or samples.dim() == 3
 
     samples = samples.detach().cpu()
@@ -45,14 +49,14 @@ def save_audio(samples: torch.Tensor):
 
     for idx, audio in enumerate(samples):
         audio_path = os.path.join(save_path, f"audio_{idx}.wav")
-        torchaudio.save(audio_path, audio, sample_rate)
-
-def get_binary_file_downloader_html(bin_file, file_label='File'):
-    with open(bin_file, 'rb') as f:
-        data = f.read()
-    bin_str = base64.b64encode(data).decode()
-    href = f'<a href="data:application/octet-stream;base64,{bin_str}" download="{os.path.basename(bin_file)}">Download {file_label}</a>'
-    return href
+        try:
+            torchaudio.save(audio_path, audio, sample_rate)
+        except Exception as e:
+            st.error(f"Error saving audio file: {e}")
+            return None
+
+    return save_path
+
 
 st.set_page_config(
     page_icon= "musical_note",