smtsead commited on
Commit
dcff8f4
·
verified ·
1 Parent(s): f8e982a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def text2story(text):
37
  str: Generated story suitable for kids aged 3-10, within 100 words.
38
  """
39
  # Add a prompt to ensure the story is happy, fun, and appropriate for kids
40
- prompt = f"Write a happy,fun and meaningful story with a clear beginning, middle, and end for kids aged 3-10 based on the following scenario: {text}."
41
 
42
  # Load the text generation model
43
  story_generator = pipeline("text-generation", model="aspis/gpt2-genre-story-generation")
 
37
  str: Generated story suitable for kids aged 3-10, within 100 words.
38
  """
39
  # Add a prompt to ensure the story is happy, fun, and appropriate for kids
40
+ prompt = f"Write a happy,fun and meaningful story for kids based {text}."
41
 
42
  # Load the text generation model
43
  story_generator = pipeline("text-generation", model="aspis/gpt2-genre-story-generation")