code: def get_batch(batch_size, model_name):
      url = "http://images.cocodataset.org/val2017/000000039769.jpg"
      image = Image.open(requests.get(url, stream=True).raw)
      feature_extractor = AutoFeatureExtractor.from_pretrained(
          model_name
      )
  
      input_features = feature_extractor(
          [image for _ in range(batch_size)], return_tensors="pt"
      )
      return input_features
  
file_path: optimum/tests/benchmark/benchmark_bettertransformer_vit.py
imports: import argparse
  import requests
  import torch
  from PIL import Image
  from transformers import AutoFeatureExtractor, AutoModel
  from optimum.bettertransformer import BetterTransformer
label_desc: Use image node_id 2 for calling a global function with example usage: get_batch(batch_size, model_name) and returns: input_features