code: def sample_retr_image(batch):
      """Sample image from similar sample by tfidf."""
  
      crops = []
      for img in batch["encoder_input_image"]:
          crops += [inception_crop(img, area_min=10, area_max=80)]
      batch["retr_images"] = np.stack(crops, axis=0)
      return batch
  
imports: import functools
  from big_vision.datasets.imagenet import class_names
  import numpy
  from scenic.dataset_lib import dataset_utils
  from scenic.dataset_lib.big_transfer import registry
  from scenic.dataset_lib.big_transfer.preprocessing import utils
  from scenic.projects.t5 import tokenizer
  import tensorflow
label_desc: Use image node_id 7 for calling a global function with example usage: sample_retr_image(batch) and returns: batch