orzhan commited on
Commit
2adce53
·
1 Parent(s): 149af33

Update app.py

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