koey811 commited on
Commit
1cdf484
·
verified ·
1 Parent(s): c7a3c41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def generate_caption(image):
27
 
28
  def generate_story(caption):
29
  # Generate the story based on the caption using the GPT-2 model
30
- prompt = f"Once upon a time, based on the image described as '{caption}', here is a short and interesting story for children aged 3-10. The story is positive and happy in tone, with added imagination:\n\n"
31
  story = story_generator(prompt, max_length=500, num_return_sequences=1)[0]["generated_text"]
32
 
33
  # Extract the story text from the generated output
 
27
 
28
  def generate_story(caption):
29
  # Generate the story based on the caption using the GPT-2 model
30
+ prompt = f"Startinh with 'Once upon a time', based on the image described as '{caption}', here is a short and interesting story for children aged 3-10. The story is positive and happy in tone, with added imagination:\n\n"
31
  story = story_generator(prompt, max_length=500, num_return_sequences=1)[0]["generated_text"]
32
 
33
  # Extract the story text from the generated output