Update app.py

app.py

@@ -24,7 +24,12 @@ def text2story(text):
                               top_p=0.95, 
                               temperature=0.7)
     generation = story_text(story_prompt)
-    story_text = generation[0]['generated_text']
+    raw_output = generation[0]['generated_text']
+    
+    if raw_output.startswith(story_prompt):
+        story_text = raw_output[len(story_prompt):].strip()
+    else:
+        story_text = raw_output
     return story_text
 
 # text2audio