Update app.py

app.py

@@ -11,12 +11,14 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    story_text = ""   # to be completed
+    story_text = pipeline("text2text-generation", model="google/flan-t5-large")  
+    story_text = pipe(text)[0]['generated_text']
     return story_text
 
 # text2audio
 def text2audio(story_text):
-    audio_data = ""     # to be completed
+    audio_data =pipeline("text-to-audio", model="facebook/musicgen-stereo-large")   
+    audio_data = pipe(story_text)
     return audio_data
 
 # main part
@@ -41,17 +43,17 @@ if uploaded_file is not None:
 
     #Stage 2: Text to Story
     st.text('Generating a story...')
-    # story = text2story(scenario)
-    # st.write(story)
+    story = text2story(scenario)
+    st.write(story)
 
     #Stage 3: Story to Audio data
-    # st.text('Generating audio data...')
-    # audio_data =text2audio(story)
+    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")
+        st.audio(audio_data['audio'],
+                   format="audio/wav",
+                   start_time=0,
+                   sample_rate = audio_data['sampling_rate'])
+        # st.audio("kids_playing_audio.wav")