From 80208079d260a7bc13a1d060fe0ec5d74ce89edf Mon Sep 17 00:00:00 2001 From: MediaPipe Team Date: Fri, 16 Jun 2023 16:43:44 -0700 Subject: [PATCH] Use `GFile` for internal file systems. PiperOrigin-RevId: 541041972 --- .../python/text/text_classifier/text_classifier.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/mediapipe/model_maker/python/text/text_classifier/text_classifier.py b/mediapipe/model_maker/python/text/text_classifier/text_classifier.py index c3dd48be8..a6762176b 100644 --- a/mediapipe/model_maker/python/text/text_classifier/text_classifier.py +++ b/mediapipe/model_maker/python/text/text_classifier/text_classifier.py @@ -161,9 +161,8 @@ class TextClassifier(classifier.Classifier): path is {self._hparams.export_dir}/{model_name}. quantization_config: The configuration for model quantization. """ - if not tf.io.gfile.exists(self._hparams.export_dir): - tf.io.gfile.makedirs(self._hparams.export_dir) tflite_file = os.path.join(self._hparams.export_dir, model_name) + tf.io.gfile.makedirs(os.path.dirname(tflite_file)) metadata_file = os.path.join(self._hparams.export_dir, "metadata.json") tflite_model = model_util.convert_to_tflite( @@ -174,7 +173,7 @@ class TextClassifier(classifier.Classifier): writer = self._get_metadata_writer(tflite_model, vocab_filepath) tflite_model_with_metadata, metadata_json = writer.populate() model_util.save_tflite(tflite_model_with_metadata, tflite_file) - with open(metadata_file, "w") as f: + with tf.io.gfile.GFile(metadata_file, "w") as f: f.write(metadata_json) @abc.abstractmethod