Update tensorflow_models usage of tf.lite.interpreter to run ai-edge-litert.interpreter

PiperOrigin-RevId: 681625530

Author: ecalubaquib

official/projects/edgetpu/vision/serving/tflite_imagenet_evaluator.py

@@ -21,6 +21,10 @@
 import numpy as np
 import tensorflow as tf, tf_keras
 
+# pylint: disable=g-direct-tensorflow-import
+from tensorflow.lite.python import interpreter as tfl_interpreter
+# pylint: enable=g-direct-tensorflow-import
+
 
 @dataclasses.dataclass
 class EvaluationInput():
@@ -58,8 +62,9 @@ def evaluate_single_image(self, eval_input: EvaluationInput) -> bool:
     Returns:
       Whether the estimation is correct.
     """
-    interpreter = tf.lite.Interpreter(
-        model_content=self._model_content, num_threads=1)
+    interpreter = tfl_interpreter.Interpreter(
+        model_content=self._model_content, num_threads=1
+    )
     interpreter.allocate_tensors()
     # Get input and output tensors and quantization details.
     input_details = interpreter.get_input_details()

official/projects/movinet/tools/export_saved_model_test.py

@@ -13,13 +13,16 @@
 # limitations under the License.
 
 """Tests for export_saved_model."""
-
 from absl import flags
 import tensorflow as tf, tf_keras
 import tensorflow_hub as hub
 
+# pylint: disable=g-direct-tensorflow-import
+from tensorflow.lite.python import interpreter as tfl_interpreter
+# pylint: enable=g-direct-tensorflow-import
 from official.projects.movinet.tools import export_saved_model
 
+
 FLAGS = flags.FLAGS
 
 
@@ -120,7 +123,7 @@ def test_movinet_export_a0_stream_with_tflite(self):
     converter = tf.lite.TFLiteConverter.from_saved_model(saved_model_path)
     tflite_model = converter.convert()
 
-    interpreter = tf.lite.Interpreter(model_content=tflite_model)
+    interpreter = tfl_interpreter.Interpreter(model_content=tflite_model)
     runner = interpreter.get_signature_runner('serving_default')
 
     def state_name(name: str) -> str:

official/requirements.txt

@@ -27,3 +27,4 @@ sentencepiece
 sacrebleu
 # Projects/vit dependencies
 immutabledict
+ai-edge-litert>=1.0.1