Update app.py

app.py

@@ -10,7 +10,9 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    story_text = ""   # to be completed
+    text_to_story = pipeline("text-generation", model="isarth/distill_gpt2_story_generator")
+    # story_text = ""   # to be completed
+    story_text = text_to_story(text)[0]["generated_text"]
     return story_text
 
 # text2audio
@@ -41,14 +43,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")