Hugging Face's logo

Norm
/
ERNIE-Layout-Pytorch

Transformers
PyTorch
xlnet
Model card Files Community
1
Norm commited on
Commit
cee5d84
·
1 Parent(s): 5dcf050

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -31,7 +31,7 @@ layout = [[381, 91, 505, 115], [738, 96, 804, 122]] # all boxes are resized bet
31
  feature_extractor = ErnieFeatureExtractor()
32
 
33
  # Tokenize context & questions
34
- context_encodings = prepare_context_info(tokenizer, context, layout, add_special_tokens=False)
35
  question = "what is it?"
36
  tokenized_res = ernie_qa_tokenize(tokenizer, question, context_encodings)
37
  tokenized_res['input_ids'] = torch.tensor([tokenized_res['input_ids']]).to(device)
 
31
  feature_extractor = ErnieFeatureExtractor()
32
 
33
  # Tokenize context & questions
34
+ context_encodings = prepare_context_info(tokenizer, context, layout)
35
  question = "what is it?"
36
  tokenized_res = ernie_qa_tokenize(tokenizer, question, context_encodings)
37
  tokenized_res['input_ids'] = torch.tensor([tokenized_res['input_ids']]).to(device)