Update app.py

app.py

@@ -12,11 +12,17 @@ def img2text(img):
 
 # text2story
 def text2story(text):
+    prompt = "Once upon a time in a land far, far away"
+
     text_generation_model = pipeline("text-generation",
-                                   model="distilbert/distilgpt2")
-    story_text = text_generation_model + " Once upon a time, in a faraway land, there lived a magical creature. This creature had the power to..."
-    while len(story_text.split()) < 100:
-        story_text += " additional content to make the story longer."
+                     model="distilbert/distilgpt2")
+
+    generated_story = text_generation_model(prompt,
+                            max_length=100,
+                            num_return_sequences=1)
+
+print("The story is here......")
+print(generated_story[0]['generated_text'])
         
 # text2audio
 def text2audio(story_text):
@@ -48,7 +54,7 @@ if uploaded_file is not None:
      # stage 2
     st.text('Generating a story...')
     scenario = text2story(scenario)
-    st.write(story_text)
+    st.write(generated_story)
 
      # stage 3
     st.text('Generating audio data...')