Open tflite as rb

google · Sep 15, 2024 · 3a8f092 · 3a8f092
1 parent 55f9e1e
commit 3a8f092
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lingvo/setup.py b/lingvo/setup.py
@@ -2,7 +2,7 @@
 
 import setuptools
 
-VERSION = "0.0.6"
+VERSION = "0.0.7"
 
 with open("README.md", "r") as file_object:
     LONG_DESCRIPTION = file_object.read()

diff --git a/lingvo/sidlingvo/fe_utils.py b/lingvo/sidlingvo/fe_utils.py
@@ -74,7 +74,7 @@ def normalize_features(
 def load_tflite_model(model_path: str) -> tf.lite.Interpreter:
   """Reads a serialized TFLite model and returns the Interpreter object."""
   # Read model.
-  with open(model_path) as file_object:
+  with open(model_path, "rb") as file_object:
     model_content = file_object.read()
   tflite_model = tf.lite.Interpreter(model_content=model_content)
   tflite_model.allocate_tensors()