Use tf.io.gfile instead of adding an explicit import for gfile. This is consistent with the examples/
Use tf.io.gfile instead of adding an explicit import for gfile. This is consistent with the examples/

google/flax,google/flax