Update README.md

README.md

@@ -26,12 +26,6 @@ layout = [[381, 91, 505, 115], [738, 96, 804, 122]]
 # intialize feature extractor
 feature_extractor = ErnieFeatureExtractor()
 
-# open the image of the document
-pil_image = Image.open("/path/to/image").convert("RGB")
-
-# Process image
-tokenized_res['pixel_values'] = feature_extractor(pil_image)
-
 # Tokenize context & questions
 context_encodings, = prepare_context_info(tokenizer, context, layout)
 question = "what is it?"
@@ -41,6 +35,13 @@ tokenized_res = ernie_qa_tokenize(tokenizer, question, context_encodings)
 tokenized_res['start_positions'] = 6
 tokenized_res['end_positions'] = 12
 
+# open the image of the document
+pil_image = Image.open("/path/to/image").convert("RGB")
+
+# Process image
+tokenized_res['pixel_values'] = feature_extractor(pil_image)
+
+
 # initialize config
 config = ErnieLayoutConfig.from_pretrained(pretrained_model_name_or_path=pretrain_torch_model_or_path)
 config.num_classes = 2 # start and end