BienKieu commited on
Commit
62d68a0
·
verified ·
1 Parent(s): ed32b22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -29,7 +29,9 @@ def summarize_one(lang: str, desc: str, code: str):
29
  max_length=GEN_MAX_LENGTH,
30
  num_beams=5,
31
  num_return_sequences=5,
32
- early_stopping=True,
 
 
33
  )
34
 
35
  summaries = [tokenizer.decode(o, skip_special_tokens=True).strip() for o in outputs]
 
29
  max_length=GEN_MAX_LENGTH,
30
  num_beams=5,
31
  num_return_sequences=5,
32
+ min_length=4,
33
+ max_length=64,
34
+ length_penalty=0.0
35
  )
36
 
37
  summaries = [tokenizer.decode(o, skip_special_tokens=True).strip() for o in outputs]