Update app.py

app.py

@@ -17,6 +17,17 @@ def text2story(text):
     
     # Create a prompt for the story generation
     prompt = f"Write a fun children's story based on this: {text}. Once upon a time, "
+    
+    # Generate the story
+    story_result = generator(
+        prompt,
+        max_length=150,
+        num_return_sequences=1,
+        temperature=0.7,
+        top_k=50,
+        top_p=0.95,
+        do_sample=True
+    )
    
     # Extract the generated text
     story_text = story_result[0]['generated_text']