destiratnakomala commited on
Commit
b70a0c8
1 Parent(s): 6db5daf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -39,9 +39,9 @@ ARTICLE_TO_SUMMARIZE = ""
39
  # generate summary
40
  input_ids = tokenizer.encode(ARTICLE_TO_SUMMARIZE, return_tensors='pt')
41
  summary_ids = model.generate(input_ids,
42
- min_length=20,
43
- max_length=80,
44
- num_beams=10,
45
  repetition_penalty=2.5,
46
  length_penalty=1.0,
47
  early_stopping=True,
 
39
  # generate summary
40
  input_ids = tokenizer.encode(ARTICLE_TO_SUMMARIZE, return_tensors='pt')
41
  summary_ids = model.generate(input_ids,
42
+ min_length=200,
43
+ max_length=800,
44
+ num_beams=100,
45
  repetition_penalty=2.5,
46
  length_penalty=1.0,
47
  early_stopping=True,