Update README.md

README.md

@@ -27,13 +27,24 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 tokenizer = AutoTokenizer.from_pretrained("preechanon/mt5-base-thaisum-text-summarization")
 model = AutoModelForSeq2SeqLM.from_pretrained("preechanon/mt5-base-thaisum-text-summarization")
+new_input_string = "ข้อความที่ต้องการสรุป"
+input_ =tokenizer(new_input_string, truncation=True, max_length=1024, return_tensors="pt")
+with torch.no_grad():
+    preds = model.generate(
+        input_['input_ids'].to('cpu'),
+        num_beams=15,
+        num_return_sequences=1,
+        no_repeat_ngram_size=1,
+        remove_invalid_values=True,
+        max_length=140,
+    )
 ```
 
 ### Score
-- Rouge1: 0.478420			
-- Rouge2: 0.308502
-- Rougel: 0.392542
-- Rougelsum: 0.434619
+- Rouge1: 0.488931			
+- Rouge2: 0.309732
+- Rougel: 0.425490
+- Rougelsum: 0.444359
 
 ### Training hyperparameters