test.py

import tensorflow as tf

# Load your H5 model (using custom_objects if necessary)
custom_objects = {"KerasLayer": tf.keras.layers.Layer}  # adjust if needed
model = tf.keras.models.load_model("models/sugercane_model.h5", custom_objects=custom_objects, compile=False)

# Save in the SavedModel format
tf.keras.models.save_model(model, "models/sugercane_model_saved", save_format="tf")