import tensorflow as tf

def check_tflite(model_path):
    interpreter = tf.lite.Interpreter(model_path=model_path)
    interpreter.allocate_tensors()

    input_details = interpreter.get_input_details()
    output_details = interpreter.get_output_details()

    print("--- Input Details ---")
    for detail in input_details:
        print(f"Name: {detail['name']}, Shape: {detail['shape']}, Type: {detail['dtype']}, Quantization: {detail['quantization']}")

    print("\n--- Output Details ---")
    for detail in output_details:
        print(f"Name: {detail['name']}, Shape: {detail['shape']}, Type: {detail['dtype']}, Quantization: {detail['quantization']}")

if __name__ == "__main__":
    check_tflite('mobile/assets/best.tflite')