Update app.py

app.py

@@ -10,14 +10,14 @@ def img2text(url):
     return text
 
 # text2story
-story_prompt = (
-    f"You are an expert in storytelling for children ages 3-10."
-    f"Based on the image description below:{text}\n"
-    f"Please create a funny, heartwarming, imaginative fairy tale of no less than 50 words"
-    f"Output only the body of the story, do not include other content"
-    )
-
 def text2story(text):
+    story_prompt = (
+        f"You are an expert in storytelling for children ages 3-10."
+        f"Based on the image description below:{text}\n"
+        f"Please create a funny, heartwarming, imaginative fairy tale of no less than 50 words"
+        f"Output only the body of the story, do not include other content"
+    )
+    
     story_text = pipeline("text-generation", model="distilbert/distilgpt2",
                               max_length=200, 
                               do_sample=True,