Update app.py

app.py

@@ -17,8 +17,9 @@ def text2story(text):
 
 # text2audio
 def text2audio(story_text):
-    audio_data = ""     # to be completed
-    return audio_data
+    audio_data = pipeline("text-to-audio", model="facebook/musicgen-medium")
+    audio_data = audio_data(story_text)
+    return audio_data
     
 st.set_page_config(page_title="Your Image to Audio Story",
                    page_icon="🦜")
@@ -43,14 +44,14 @@ if uploaded_file is not None:
     story = text2story(scenario)
     st.write(story)
 
-    # #Stage 3: Story to Audio data
-    # st.text('Generating audio data...')
-    # audio_data =text2audio(story)
-
-    # # Play button
-    # if st.button("Play Audio"):
-    #     st.audio(audio_data['audio'],
-    #                format="audio/wav",
-    #                start_time=0,
-    #                sample_rate = audio_data['sampling_rate'])
-    #     # st.audio("kids_playing_audio.wav")
+    #Stage 3: Story to Audio data
+    st.text('Generating audio data...')
+    audio_data =text2audio(story)
+
+    # Play button
+    if st.button("Play Audio"):
+        st.audio(audio_data['audio'],
+                   format="audio/wav",
+                   start_time=0,
+                   sample_rate = audio_data['sampling_rate'])
+        # st.audio("kids_playing_audio.wav")