small updates

MusicModel/music.py

@@ -12,7 +12,8 @@ import streamlit as st
 # Spits out audio file
 
 def gen_audio():
+    to_ret = ""
     for i in range(5):
-        st.write(str(i))
+        to_ret = to_ret + str(i)
 
-    st.write("Generation Complete")
+    return to_ret

app.py

@@ -13,4 +13,4 @@ st.write("Click the Button to Generate Audio")
 # SENTENCE BASED 
 if st.button("GENERATE"):
     st.write("Here is the Generated Audio:")
-    gen_audio()
+    st.write(gen_audio())