Update README.md

README.md

@@ -29,11 +29,13 @@ model = AutoModelForTokenClassification.from_pretrained(nielsr/layoutxlm-finetun
 image = Image("...").convert("RGB")
 
 # prepare for the model
-encoding = processor(image, return_tensors="pt")
+encoding = processor(image, return_offsets_mapping=True, padding="max_length", max_length=512, truncation=True, return_tensors="pt")
 
 with torch.no_grad():
   outputs = model(**encoding)
   logits = outputs.logits
+  
+predictions = logits.argmax(-1)
 ```
 
 ## Intended uses & limitations