code: def __init__(self):
      super(MyModel, self).__init__()
      self.conv1 = Conv2D(32, 3, activation="relu")
      self.flatten = Flatten()
      self.d1 = Dense(128, activation="relu")
      self.d2 = Dense(10)
  
file_path: BentoML/examples/tensorflow2_keras/train.py
imports: import tensorflow
  from tensorflow.keras import Model
  from tensorflow.keras.layers import Conv2D
  from tensorflow.keras.layers import Dense
  from tensorflow.keras.layers import Flatten
  import bentoml
label_desc: Use image node_id 1 to create a new MyModel object from inherited base classes: Model with example: obj = MyModel()