Update function.py

function.py

@@ -27,7 +27,7 @@ def text2story(text):
     model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
 
     prompt = (
-        f"Please write a short, imaginative story based on the following idea (no more than 100 words):\n\n"
+        f"Please write a short, imaginative story based on the following idea (no more than 10 words):\n\n"
         f"{text}\n\nStory:"
     )
     inputs = tokenizer(prompt, return_tensors="pt")