Update app.py

app.py

@@ -6,12 +6,8 @@ from transformers import pipeline
 toxic_model = pipeline("text-classification", model="Matt09Miao/GP5_tweet_toxic")  
     
 
-# text2story
-def text2story(input):
-    pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
-    tweet = pipe(input)
-    return tweet
-
+text2tweet = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
+ 
 # text2audio
 def text2audio(toxic_result):
     pipe = pipeline("text-to-audio", model="Matthijs/mms-tts-eng")
@@ -28,7 +24,7 @@ input = st.text_input("Please input your first word...")
 if input is not None:
     #Stage 1: Input to Tweet
     st.text('Generating a Tweet...')
-    tweet = text2story(input)
+    tweet = text2tweet(input)
     st.write(tweet)
     
     #Stage 2: Tweet Toxicity Analysis