Fix code example

README.md

@@ -36,10 +36,9 @@ processor = TrOCRProcessor.from_pretrained('microsoft/trocr-base-stage1')
 model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-base-stage1')
 
 # training
-pixel_values = processor(images=image, return_tensors="pt").pixel_values
-labels = processor.tokenizer('industrie', return_tensors='pt').input_ids
-outputs = model(pixel_values=pixel_values, labels=decoder_input_ids)
-loss = outputs.loss
+pixel_values = processor(image, return_tensors="pt").pixel_values  # Batch size 1
+decoder_input_ids = torch.tensor([[model.config.decoder.decoder_start_token_id]])
+outputs = model(pixel_values=pixel_values, decoder_input_ids=decoder_input_ids)
 ```
 
 ### BibTeX entry and citation info