MODEL_DIR = "/Users/amit/Desktop/test/exportpath"
version = 1
export_path = os.path.join(MODEL_DIR, str(version))
print('export_path = {}\n'.format(export_path))
if os.path.isdir(export_path):
  print('\nAlready saved a model, cleaning up\n')
with tf.Session(graph=graph) as sess:
    tf.saved_model.simple_save(
        sess,
        export_path,
        inputs={'input_image': x},
        outputs={'final_output':y})

print('\nSaved model:')