code: def __init__(self, model):  # skipcq: PYL-W0621
      self.model = model
  
imports: from pathlib import Path
  import pytorch_lightning
  import torch
  from flash.core.serve import ModelComponent, expose
  from flash.core.serve.types import Image, Label, Number, Repeated
  from flash.core.utilities.imports import _TORCHVISION_AVAILABLE
  from torch import Tensor
label_desc: Use image node_id 1 to create a new ClassificationInference object from inherited base classes: ModelComponent with example: obj = ClassificationInference(model)