Move tensorflow lite python calls to ai-edge-litert.

PiperOrigin-RevId: 684104608

chirp/projects/zoo/models.py

@@ -33,6 +33,8 @@
 import tensorflow.compat.v1 as tf1
 import tensorflow_hub as hub
 
+from ai_edge_litert import interpreter as tfl_interpreter  # pylint: disable=g-direct-tensorflow-import
+
 
 def model_class_map() -> dict[str, Any]:
   """Get the mapping of model keys to classes."""
@@ -419,7 +421,7 @@ def from_config(cls, config: config_dict.ConfigDict) -> 'BirdNet':
       with tempfile.NamedTemporaryFile() as tmpf:
         model_file = epath.Path(config.model_path)
         model_file.copy(tmpf.name, overwrite=True)
-        model = tf.lite.Interpreter(
+        model = tfl_interpreter.Interpreter(
             tmpf.name, num_threads=config.num_tflite_threads
         )
       model.allocate_tensors()

chirp/train_tests/frontend_test.py

@@ -23,6 +23,7 @@
 
 from absl.testing import absltest
 from absl.testing import parameterized
+from ai_edge_litert import interpreter as tfl_interpreter  # pylint: disable=g-direct-tensorflow-import
 
 
 class FrontendTest(parameterized.TestCase):
@@ -200,7 +201,7 @@ def test_tflite_stft_export(
     tflite_float_model = converter.convert()
 
     # Use the converted TFLite model.
-    interpreter = tf.lite.Interpreter(model_content=tflite_float_model)
+    interpreter = tfl_interpreter.Interpreter(model_content=tflite_float_model)
     interpreter.allocate_tensors()
     input_tensor = interpreter.get_input_details()[0]
     output_tensor = interpreter.get_output_details()[0]
@@ -247,7 +248,7 @@ def test_simple_melspec(self):
     tflite_float_model = converter.convert()
 
     # Use the converted TFLite model.
-    interpreter = tf.lite.Interpreter(model_content=tflite_float_model)
+    interpreter = tfl_interpreter.Interpreter(model_content=tflite_float_model)
     interpreter.allocate_tensors()
     input_tensor = interpreter.get_input_details()[0]
     output_tensor = interpreter.get_output_details()[0]