code: def __init__(self, model_name):
      self.tokenizer = AutoTokenizer.from_pretrained(model_name)
      self.model = BertModel.from_pretrained(model_name).eval()
      self.model.cuda()
  
imports: import argparse
  import numpy
  import torch
  from tqdm import tqdm
label_desc: Use image node_id 1 to create a new BertFeatExtractor object  with example: obj = BertFeatExtractor(model_name)