orzhan commited on
Commit
1fe7ac7
·
1 Parent(s): 661df71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,8 +20,8 @@ def infer(input_ids):
20
  # do_sample=False,
21
  num_return_sequences=1,
22
  num_beams=8,
23
- length_penalty=0.5,
24
- no_repeat_ngram_size=4,
25
  # do_sample=True, top_p=0.9, temperature=0.9
26
  )
27
 
 
20
  # do_sample=False,
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