Update README.md

README.md

@@ -10,8 +10,8 @@ license: cc-by-sa-4.0
 datasets:
 - xsum
 widget:
-- text: "<s> Ban Ki-moon was elected for a second term in 2007 </s></s> Ban-Ki Moon was re-elected for a second term by the UN General Assembly, unopposed and unanimously, on 21 June 2011"
-- text: "<s> Ban Ki-moon was elected for a second term in 2011 </s></s> Ban-Ki Moon was re-elected for a second term by the UN General Assembly, unopposed and unanimously, on 21 June 2011"
+- text: "<s> Ban Ki-moon was elected for a second term in 2007. </s></s> Ban Ki-Moon was re-elected for a second term by the UN General Assembly, unopposed and unanimously, on 21 June 2011."
+- text: "<s> Ban Ki-moon was elected for a second term in 2011. </s></s> Ban Ki-Moon was re-elected for a second term by the UN General Assembly, unopposed and unanimously, on 21 June 2011."
 
 ---
 
@@ -31,10 +31,10 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 tokenizer = AutoTokenizer.from_pretrained("CogComp/bart-faithful-summary-detector")
 model = AutoModelForSequenceClassification.from_pretrained("CogComp/bart-faithful-summary-detector")
 
-article = "Ban Ki-Moon was re-elected for a second term by the UN General Assembly, unopposed and unanimously, on 21 June 2011"
+article = "Ban Ki-Moon was re-elected for a second term by the UN General Assembly, unopposed and unanimously, on 21 June 2011."
 
-bad_summary = "Ban Ki-moon was elected for a second term in 2007"
-good_summary = "Ban Ki-moon was elected for a second term in 2011"
+bad_summary = "Ban Ki-moon was elected for a second term in 2007."
+good_summary = "Ban Ki-moon was elected for a second term in 2011."
 
 bad_pair = tokenizer(text=bad_summary, text_pair=article, return_tensors='pt')
 good_pair = tokenizer(text=good_summary, text_pair=article, return_tensors='pt')