Vijayendra commited on
Commit
3e5c877
·
verified ·
1 Parent(s): 1038f41

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -37,7 +37,7 @@ misinformation policies on YouTube with new guidelines on currently administered
37
  effective by local health authorities and the WHO," the post said, referring to the World Health Organization.
38
  """
39
  inputs = tokenizer.encode("summarize: " + article, return_tensors="pt", max_length=512, truncation=True)
40
- summary_ids = model.generate(inputs, max_length=150, min_length=50, length_penalty=2.0, num_beams=4, early_stopping=True)
41
 
42
  # Decode and print the summary
43
  summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
 
37
  effective by local health authorities and the WHO," the post said, referring to the World Health Organization.
38
  """
39
  inputs = tokenizer.encode("summarize: " + article, return_tensors="pt", max_length=512, truncation=True)
40
+ summary_ids = model.generate(inputs, max_length=150, min_length=100, length_penalty=2.0, num_beams=4, early_stopping=True)
41
 
42
  # Decode and print the summary
43
  summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)