Update app.py

app.py

@@ -15,7 +15,7 @@ def text2story(text):
     - Beginning, middle, end
     - Character development
     - At least 3 story elements"""
-    text_to_story_model= pipeline("text2text-generation", model="google/flan-t5-small")
+    text_to_story_model= pipeline("text-generation", model="mosaicml/mpt-7b-storywriter", trust_remote_code=True)
     story_text = text_to_story_model(
         messages,
         max_length=200,