mjpsm commited on
Commit
04f8f11
·
verified ·
1 Parent(s): 066622d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -24,7 +24,9 @@ def generate_affirmation(description):
24
  top_k=50,
25
  top_p=0.95,
26
  do_sample=True,
27
- pad_token_id=tokenizer.eos_token_id
 
 
28
  )
29
 
30
  full_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
24
  top_k=50,
25
  top_p=0.95,
26
  do_sample=True,
27
+ pad_token_id=tokenizer.eos_token_id,
28
+ repetition_penalty=1.2,
29
+ no_repeat_ngram_size=3
30
  )
31
 
32
  full_output = tokenizer.decode(outputs[0], skip_special_tokens=True)