Update app.py

app.py

@@ -37,11 +37,11 @@ def main():
     title = st.text_input('Write a prompt (จะใช้เวลาค่อนข้างมากในการสร้างเนื่องจากใช้ CPU ในการรันโมเดล)', "")
 
     if st.button('Generate Image'):
-        # Call the pic_mo function to generate an image
-        generated_music = mu_gen(prompt)
+        # Call the mu_gen function to generate music using the 'title' prompt
+        generated_music = mu_gen(title)  # Replace 'prompt' with 'title'
 
-        # Display the generated image
-        st.image(generated_music, caption='Generated Music', use_column_width=True)
+        # Display the generated audio
+        st.audio(generated_music, format='audio/mpeg', start_time=0)
 
 if __name__ == '__main__':
-    main()
+    main()