orzhan commited on
Commit
d117191
·
1 Parent(s): db6b790

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -21,7 +21,8 @@ def infer(input_ids):
21
  num_return_sequences=1,
22
  num_beams=8,
23
  length_penalty=1.0,
24
- no_repeat_ngram_size=4
 
25
  )
26
 
27
  return output_sequences
 
21
  num_return_sequences=1,
22
  num_beams=8,
23
  length_penalty=1.0,
24
+ no_repeat_ngram_size=4,
25
+ do_sample=True, top_p=0.9, temperature=0.9
26
  )
27
 
28
  return output_sequences