updates

app.py

@@ -6,6 +6,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipe
 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')
 
 
 with st.form("my_form"):
@@ -36,9 +37,8 @@ with st.form("my_form"):
         
         if out[0]["label"] == "POSITIVE" or out[0]["label"] == "POS":
             # 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!"
-            # generator = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
-            # response = generator(prompt)
+            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")
         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!"