code: def __init__(
      self,
      serve_input: "ServeInput",
      collate_fn: Callable,
  ):
      super().__init__()
      self.serve_input = serve_input
      self.collate_fn = collate_fn
  
file_path: lightning-flash/src/flash/core/data/batch.py
imports: from typing import TYPE_CHECKING, Any, Callable, List
  from torch import Tensor, nn
  from flash.core.data.utilities.classification import _is_list_like
label_desc: Use image node_id 1 to create a new _ServeInputProcessor object from inherited base classes: nn with example: obj = _ServeInputProcessor(serve_input, collate_fn)