updates

app.py

@@ -35,13 +35,14 @@ 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)
-            st.error(response)
+            # 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)
+            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!"
-            generator = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
-            response = generator(prompt)
-            st.success(response)
+            # 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")