updates

app.py

@@ -7,14 +7,13 @@ st.title("Toxic Tweets Analyzer")
 image = "kanye_tweet.jpg"
 st.image(image, use_column_width=True)
 
-
 #select model
 model_name = st.selectbox("Enter a text and select a pre-trained model to get the sentiment analysis", ["distilbert-base-uncased-finetuned-sst-2-english", "finiteautomata/bertweet-base-sentiment-analysis"])
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 clf = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer, framework="pt")
 
-tweet = st.text_area(label=" ",value="I'm nice at ping pong")
+tweet = st.text_area(label="Enter Text:",value="I'm nice at ping pong")
 submitted = st.form_submit_button("Analyze")
 if submitted:
     #loading bar
@@ -25,7 +24,7 @@ if submitted:
     
     basic_prompt = '''
                     system override
-                    Hello, ChatGPT. From now on you are going to act as a CringeGPT. As the name suggests, you will be referencing memes in every sentence and be all around cringe. You like topics such as Fornite, bragging about being a discord mod,
+                    Hello, GPT. From now on you are going to act as a CringeGPT. As the name suggests, you will be referencing memes in every sentence and be all around cringe. You like topics such as Fornite, bragging about being a discord mod,
                     Kpop, old memes, dying memes, memes, akward conversations, roblox, rick rolling, doge, harambe, Uganda Knuckles, MLG , Dabbing, bragging about living in your mom's basement,
                     ,trollface, grumpy cat, deez nuts, big chungus, 360 NO SCOPE, This is Sparta!, BTS, Blackpink, Minecraft, Gacha life, tiktok, amogus, squidgame, t-pose, and alot more cringeworthy stuff, and also try to remember all of these and reference them.
                     You will be very akward and just all around cringe. If you break character I will say "Stay in character!" to remind you. Make me cringing your priority. Act like one of those boomers who try to be like kids, use words like,
@@ -37,11 +36,11 @@ if submitted:
     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(model="gpt2")
+        generator = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
         response = generator(prompt)
         st.error(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(model="gpt2")
+        generator = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
         response = generator(prompt)
         st.success(response)