Update app.py

app.py

@@ -80,8 +80,6 @@ def main():
     left_co.write("""Music Generation through a prompt""")
     left_co.write(("""PS : First generation may take some time ......."""))
 
-save_path = "audio_output/"
-os.makedirs(save_path, exist_ok=True)  # Create the directory if it doesn't exist
 
 if st.sidebar.button('Generate !'):
      with left_co:
@@ -110,6 +108,10 @@ if st.sidebar.button('Generate !'):
          st.audio(audio_bytes, format='audio/wav')
          st.markdown(get_binary_file_downloader_html(audio_filepath, 'Audio'), unsafe_allow_html=True)
 
+save_path = "audio_output/"
+os.makedirs(save_path, exist_ok=True)  # Create the directory if it doesn't exist
+
+
 
 if __name__ == "__main__":
     main()