code: def main(unused_argv=None):
      # Load image
      image = np.expand_dims(
          image_utils.load_np_image(
              os.path.expanduser(FLAGS.input_image)
          ),
          0,
      )
  
      output_dir = os.path.expanduser(FLAGS.output_dir)
      if not os.path.exists(output_dir):
          os.makedirs(output_dir)
  
      which_styles = ast.literal_eval(FLAGS.which_styles)
      if isinstance(which_styles, list):
          _multiple_images(image, which_styles, output_dir)
      elif isinstance(which_styles, dict):
          _multiple_styles(image, which_styles, output_dir)
      else:
          raise ValueError(
              "--which_styles must be either a list of style indexes "
              "or a dictionary mapping style indexes to weights."
          )
  
file_path: magenta/magenta/models/image_stylization/image_stylization_transform.py
imports: import ast
  import os
  from magenta.models.image_stylization import image_utils
  from magenta.models.image_stylization import model
  from magenta.models.image_stylization import ops
  import numpy
  import tensorflow.compat.v1
label_desc: Use image node_id 6 for calling a global function with example usage: main(unused_argv) without return types