Update app.py

app.py

@@ -10,7 +10,7 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    text_to_story = pipeline("text2text-generation", model="Salesforce/codet5p-220m-py")
+    text_to_story = pipeline("text-generation", model="mosaicml/mpt-7b-storywriter", trust_remote_code=True)
     # story_text = ""   # to be completed
     story_text = text_to_story(text)[0]["generated_text"]
     return story_text