Update app.py

app.py

@@ -8,8 +8,8 @@ toxic_model = pipeline("text-classification", model="Matt09Miao/GP5_tweet_toxic"
 
 # text2story
 def text2story(text):
-    pipe = pipeline("text-generation", model="distilbert/distilgpt2")
-    tweet_text = pipe(text)[0]['generated_text']
+    pipe = pipeline("text2text-generation", model="tweetpie/stance-directed-tweet-generator")        
+    tweet_text = pipe(text)
     return tweet_text
 
 # text2audio