Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
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")
|