why use tflite when you can use tensorflow

utils/color_model.py

@@ -1,16 +1,16 @@
 import numpy as np
+import tensorflow as tf
 from huggingface_hub import hf_hub_download
-import tflite_runtime.interpreter as tflite
 
 
-# Download TFLite model file
+# Download model file from HF
 model_path = hf_hub_download(
     repo_id="danielritchie/vibe-color-model",
-    filename="vibe_model.tflite"
+    filename="vibe_model.tflite"   # confirm exact filename
 )
 
-# Load interpreter
-interpreter = tflite.Interpreter(model_path=model_path)
+# Load TFLite interpreter
+interpreter = tf.lite.Interpreter(model_path=model_path)
 interpreter.allocate_tensors()
 
 input_details = interpreter.get_input_details()