smtsead commited on
Commit
8ac8023
·
verified ·
1 Parent(s): a0cc8a7

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 complete story for kids based on the following scenario: {text} with no sad, violent, scary elements, guns, weapons, or attacks."
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 complete story for kids based on the following scenario: {text} with no sad, violent, scary elements."
41
 
42
  # Load the text generation model
43
  story_generator = pipeline("text-generation", model="aspis/gpt2-genre-story-generation")