import tensorflow as tf

    export_path = image_classifier.export_saved_model(
        export_dir_base=FLAGS.export_dir,
        serving_input_receiver_fn=build_image_serving_input_receiver_fn(
            serving_shape),
        as_text=True)
    if FLAGS.add_warmup_requests:
      write_warmup_requests(
          export_path,
          FLAGS.model_name,
          hparams.image_size,
          batch_sizes=FLAGS.inference_batch_sizes)


if __name__ == '__main__':
  tf.logging.set_verbosity(tf.logging.INFO)
  app.run(main)
