Update app.py

app.py

@@ -1,25 +1,24 @@
+import streamlit as st
+from transformers import pipeline
+
 # function part
 # Toxicity Analysis
-def toxic(text_tweet):
-    image_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
-    text = image_to_text_model(url)[0]["generated_text"]
-    return text
+    toxic_model= pipeline("text-classification", model="Matt09Miao/GP5_tweet_toxic",return_all_scores=True))    
+   
 
 # text2story
 def text2story(text):
     pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
-    story_text = pipe(text)[0]['generated_text']
-    return story_text
+    tweet_text_text = pipe(text)[0]['generated_text']
+    return tweet_text
 
 # text2audio
-def text2audio(story_text):
+def text2audio(toxic_result):
     pipe = pipeline("text-to-audio", model="Matthijs/mms-tts-eng")
-    audio_data = pipe(story_text)
+    audio_data = pipe(toxic_result)
     return audio_data
 
 
-import streamlit as st
-from transformers import pipeline
 
 st.set_page_config(page_title="Generate Your Tweet and Toxicity Analysis")