updates

app.py

@@ -21,7 +21,6 @@ if submitted:
     with st.spinner(text="loading..."):
         out = clf(tweet)
     st.success('Done!')
-    st.balloons()
     st.json(out)
     
     basic_prompt = '''
@@ -36,6 +35,7 @@ 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")
         response = generator(prompt)