Update README.md

README.md

@@ -16,12 +16,12 @@ pip install transformers
 
 # Example
 ```python
-from transformers import PegasusForConditionalGeneration, AutoTokenizer
+from transformers import PegasusXForConditionalGeneration, AutoTokenizer
 
-model = PegasusForConditionalGeneration.from_pretrained("satjawat/pegasus-x-thai-sum")
+model = PegasusXForConditionalGeneration.from_pretrained("satjawat/pegasus-x-thai-sum")
 tokenizer = AutoTokenizer.from_pretrained("satjawat/pegasus-x-thai-sum")
 
-new_input_string = "ข้อความทดสอบ"
+new_input_string = "ข้อความ"
 new_input_ids = tokenizer(new_input_string.lower(), return_tensors="pt").input_ids
 summary_ids = model.generate(new_input_ids, max_length=50, num_beams=6, length_penalty=2.0, early_stopping=True)
 summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)