code: def __init__(self):
      super().__init__()
      import torch
  
      print("Running on device:", self.device_id)
      self.torch_version = torch.__version__
      print("Running on torch version:", self.torch_version)
  
file_path: BentoML/examples/custom_model_runner/service.py
imports: from __future__ import annotations
  from typing import TYPE_CHECKING
  import time
  import numpy
  from utils import exponential_buckets
  import bentoml
label_desc: Use image node_id 1 to create a new CustomMnistRunnable object from inherited base classes: _BuiltinRunnable with example: obj = CustomMnistRunnable()