code: def load_sample(self, sample: Dict[str, Any]) -> Dict[str, Any]:
      w, h = sample[DataKeys.INPUT].size  # W x H
      if DataKeys.METADATA not in sample:
          sample[DataKeys.METADATA] = {}
      sample[DataKeys.METADATA].update(
          {
              "size": (h, w),
              "height": h,
              "width": w,
          }
      )
      return sample
  
imports: import base64
  from io import BytesIO
  from pathlib import Path
  from typing import Any, Dict, List
  import torch
  import flash
  from flash.core.data.io.input import DataKeys, Input, ServeInput
  from flash.core.data.utilities.loading import IMG_EXTENSIONS, NP_EXTENSIONS, load_image
  from flash.core.data.utilities.paths import PATH_TYPE, filter_valid_files
  from flash.core.data.utilities.samples import to_samples
  from flash.core.utilities.imports import _TORCHVISION_AVAILABLE, Image, requires
label_desc: Use image node_id 1 for calling the ImageInput obj's underlying member method code with example usage: obj.load_sample(sample) and returns: sample