updates

app.py

@@ -5,7 +5,9 @@ from transformers import AutoModelForSequenceClassification, pipeline, AutoToken
 st.title("Toxic Tweets Analyzer")
 image = "kanye_tweet.jpg"
 st.image(image, use_column_width=True)
-generator = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
+
+
+generator = pipeline('text-generation', model='gpt2')
 
 
 with st.form("my_form"):
@@ -35,13 +37,11 @@ with st.form("my_form"):
                         '''
         
         if out[0]["label"] == "POSITIVE" or out[0]["label"] == "POS":
-            # st.balloons()
+            st.balloons()
             prompt = f"{basic_prompt} + \n\nThe user wrote a tweet that says: {tweet}, compliment them on how nice of a person they are! Remember try to be as cringe and awkard as possible!"
             response = generator(prompt)
-            st.success("nice")
+            st.success(response)
         else:
-            # prompt = f"{basic_prompt} + \n\nThe user wrote a tweet that says: {tweet}, tell them on how terrible of a person they are! Remember try to be as cringe and awkard as possible!"
-            # generator = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
-            # response = generator(prompt)
-            # st.success(response)
-            st.fail("ew")
+            prompt = f"{basic_prompt} + \n\nThe user wrote a tweet that says: {tweet}, tell them on how terrible of a person they are! Remember try to be as cringe and awkard as possible!"
+            response = generator(prompt)
+            st.fail(response)