18 lines
433 B
Python
18 lines
433 B
Python
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)
|