Update app.py

app.py

@@ -74,17 +74,13 @@ st.markdown("---")
 
 st.header("Examples")
 
-example_columns = st.columns(4)
 example_titles = ["Original Audio", "Separated Audio", "AudioLDM2", "StableAudio"]
 
-for col, title in zip(example_columns, example_titles):
-    col.subheader(title)
-
 for p in exmples_prompts:
-    with st.container():
-        st.write(f"Prompt: {p}")
+    with st.expander(f"Prompt: {p}", expanded=False):
         audio_columns = st.columns(4)
-        for col in audio_columns:
+        for col, title in zip(audio_columns, example_titles):
+            col.subheader(title)
             col.audio("goodres.wav", format="audio/mpeg", loop=False)
         st.markdown("<hr style='border: 1px solid #ddd;'>", unsafe_allow_html=True)