Update README.md

README.md

@@ -24,11 +24,12 @@ This repository contains the finetuned mT5-base model for Thai sentence summariz
 ### Example
 ```python
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import torch
 
 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")
+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'),
@@ -38,6 +39,9 @@ with torch.no_grad():
         remove_invalid_values=True,
         max_length=140,
     )
+
+summary = tokenizer.decode(preds[0], skip_special_tokens=True)
+summary
 ```
 
 ### Score