import tensorflow as tf

MODEL = "models/dermy-binary-classifier"


model = tf.keras.models.load_model(MODEL + ".keras")

converter = tf.lite.TFLiteConverter.from_keras_model(model)

converter.allow_custom_ops = True
converter.experimental_enable_resource_variables = True
converter.experimental_new_converter = True

converted_model = converter.convert()

with open(MODEL + ".tflite", "wb") as file:
    file.write(converted_model)