Update app.py

app.py

@@ -6,13 +6,6 @@ import os
 import numpy as np
 import base64
 
-# Add this function for downloading binary files
-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="{file_label}">Download {file_label}</a>'
-    return href
 
 @st.cache_resource()
 def load_model():
@@ -44,10 +37,10 @@ def save_audio(samples: torch.Tensor):
     sample_rate = 30000
     save_path = "/tmp/audio_output"  # Use /tmp directory
 
-    if not os.path.exists(save_path):
-        os.makedirs(save_path)
+    # if not os.path.exists(save_path):
+    #     os.makedirs(save_path)
 
-    assert samples.dim() == 2 or samples.dim() == 3
+    # assert samples.dim() == 2 or samples.dim() == 3
 
     samples = samples.detach().cpu()
     if samples.dim() == 2:
@@ -63,9 +56,21 @@ def save_audio(samples: torch.Tensor):
 
     return save_path
 
+
+    
+
 # Define the genres list
 genres = ["Pop", "Rock", "Jazz", "Electronic", "Hip-Hop", "Classical", "Lofi", "Chillpop"]
 
+
+# Add this function for downloading binary files
+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="{file_label}">Download {file_label}</a>'
+    return href
+
 st.set_page_config(
     page_icon= "musical_note",
     page_title= "Music Gen"
@@ -109,7 +114,7 @@ def main():
             idx = 0
             music_tensor = music_tensors[idx]
             save_music_file = save_audio(music_tensor)
-            audio_filepath = f'audio_output/audio_{idx}.wav'
+            audio_filepath = f'/tmp/audio_output/audio_{idx}.wav'
             audio_file = open(audio_filepath, 'rb')
             audio_bytes = audio_file.read()