crumb commited on
Commit
c702f56
1 Parent(s): 66a1da2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -12,6 +12,7 @@ def predict(text, max_length=64):
12
  eos_token_id = tokenizer.eos_token_id,
13
  bos_token_id = tokenizer.bos_token_id,
14
  pad_token_id = tokenizer.pad_token_id,
 
15
  )[0]['generated_text']
16
  out_text = "<p>" + out_text + "</p>"
17
  out_text = out_text.replace(text, text + "<b><span>")
 
12
  eos_token_id = tokenizer.eos_token_id,
13
  bos_token_id = tokenizer.bos_token_id,
14
  pad_token_id = tokenizer.pad_token_id,
15
+ repetition_penalty = 1.05,
16
  )[0]['generated_text']
17
  out_text = "<p>" + out_text + "</p>"
18
  out_text = out_text.replace(text, text + "<b><span>")