code: def save_graph(graph, model_id):
      tf.io.write_graph(
          graph,
          "/tmp/{}".format(model_id),
          "{}/{}.pb".format(graph_path, model_id),
          False,
      )
      print("Graph exported to {}/{}.pb".format(graph_path, model_id))
  
imports: import tensorflow.compat.v1
  import tensorflow.keras
  from arguments import parse_arguments
label_desc: Use image node_id 1 for calling a global function with example usage: save_graph(graph, model_id) without return types