Update README.md

update code example

README.md

@@ -30,21 +30,21 @@ You can use the raw model for optical character recognition (OCR) on text images
 Here is how to use this model in PyTorch:
 
 ```python
-from transformers import MGPSTRProcessor, MGPSTRModel
+from transformers import MgpstrProcessor, MgpstrForSceneTextRecognition
 import requests
 from PIL import Image
 
-processor = MGPSTRProcessor.from_pretrained('alibaba-damo/mgp-str-base')
-model = MGPSTRModel.from_pretrained('alibaba-damo/mgp-str-base')
+processor = MgpstrProcessor.from_pretrained('alibaba-damo/mgp-str-base')
+model = MgpstrForSceneTextRecognition.from_pretrained('alibaba-damo/mgp-str-base')
 
 # load image from the IIIT-5k dataset
 url = "https://i.postimg.cc/ZKwLg2Gw/367-14.png"
 image = Image.open(requests.get(url, stream=True).raw).convert("RGB")
 
-pixel_values = processor(image, return_tensors="pt").pixel_values
-generated_ids, attens = model(pixel_values)
+pixel_values = processor(images=image, return_tensors="pt").pixel_values
+outputs = model(pixel_values)
 
-generated_text = processor.batch_decode(generated_ids)['generated_text']
+generated_text = processor.batch_decode(outputs.logits)['generated_text']
 ```
 
 ### BibTeX entry and citation info