code: def run_tf_model(
      model: tf.Module,
      input_tensors: Tuple[tf.Tensor],
      device: Device,
  ) -> Tuple[tf.Tensor]:
      with tf.device(device.to_tf_format()):
          pred = model(input_tensors)
      if isinstance(pred, tf.Tensor):
          pred = (pred,)
      return pred
  
file_path: nebuly/optimization/nebullvm/nebullvm/tools/tf.py
imports: from typing import Union, List, Tuple, Any, Optional, Dict
  import numpy
  from loguru import logger
  from nebullvm.core.models import Device, DataType, InputInfo
  from nebullvm.optional_modules.tensorflow import tensorflow
label_desc: Use image node_id 3 for calling a global function with example usage: run_tf_model(model, input_tensors, device) and returns: pred