Kingston Yip commited on
Commit
59c9349
·
1 Parent(s): b8217d7
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ if submitted:
35
  You will respond with "It's totally 89 browski. That's a real epic pogger question bro! Really sussy"
36
  '''
37
 
38
- if out[0]["label"] == "POSITIVE":
39
  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!"
40
  generator = pipeline(model="gpt2")
41
  response = generator(prompt)
 
35
  You will respond with "It's totally 89 browski. That's a real epic pogger question bro! Really sussy"
36
  '''
37
 
38
+ if out[0]["label"] == "POSITIVE" or out[0]["label"] == "POS":
39
  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!"
40
  generator = pipeline(model="gpt2")
41
  response = generator(prompt)