koey811 commited on
Commit
510b489
·
verified ·
1 Parent(s): 0de4b61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def generate_caption(image):
26
 
27
  def generate_story(caption):
28
  # Generate the story based on the caption using the GPT-2 model
29
- prompt = f"Once upon a time, in a world inspired by the image described as '{caption}', a delightful story unfolded for children aged 3-10. The story should be engaging, easy to understand, and use age-appropriate language. Let's begin the story:\n\n"
30
  story = story_generator(prompt, max_length=500, num_return_sequences=1)[0]["generated_text"]
31
 
32
  # Extract the story text from the generated output
 
26
 
27
  def generate_story(caption):
28
  # Generate the story based on the caption using the GPT-2 model
29
+ prompt = f"Imagine a delightful children's fairy tale inspired by the image described as '{caption}'. Itnshould be interesting, easy to understand, and use age-appropriate language suitable for children aged 3-10. Let the magical story unfold:\n\n"
30
  story = story_generator(prompt, max_length=500, num_return_sequences=1)[0]["generated_text"]
31
 
32
  # Extract the story text from the generated output