model.save('tf_model.h5')