updates

app.py

@@ -39,9 +39,9 @@ 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!"
-            response = generator(prompt)[0]
+            response = generator(prompt, max_length=1000)[0]
             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!"
-            response = generator(prompt)[0]
+            response = generator(prompt, max_length=1000)[0]
             st.error(response)