code: def create_image_model(self):
      data = tf.keras.layers.Input(shape=[224, 224, 3])
      x = tf.keras.layers.Flatten()(data)
      predictions = tf.keras.layers.Dense(10, activation="softmax")(x)
  
      model = tf.keras.models.Model(inputs=data, outputs=predictions)
      model.compile(
          optimizer="rmsprop",
          loss="sparse_categorical_crossentropy",
          metrics=["accuracy"],
      )
  
      return KerasModel(model)
  
imports: import pytest
  from pyspark.ml.linalg import DenseVector
  from bigdl.dllib.feature.common import ChainedPreprocessing, FeatureSet
  from bigdl.dllib.feature.image import *
  from bigdl.orca.test_zoo_utils import ZooTestCase
  from bigdl.dllib.keras.optimizers import Adam
  from bigdl.orca.tfpark import TFNet, TFOptimizer
  import tensorflow
  import numpy
  import os
  from bigdl.orca.tfpark import KerasModel, TFDataset
label_desc: Use image node_id 6 for calling the TestTFDataset obj's underlying member method code with example usage: obj.create_image_model() and returns: KerasModel