diff --git a/scripts/export_tvm_metadata.py b/scripts/export_tvm_metadata.py
index 335ee02..ed7871f 100644
--- a/scripts/export_tvm_metadata.py
+++ b/scripts/export_tvm_metadata.py
@@ -40,14 +40,7 @@ def sample_tflite_model() -> bytes:
     outputs = keras.layers.Dense(4, activation="softmax")(x)
     model = keras.Model(inputs=inputs, outputs=outputs)
     model.compile()
-
-    @tf.function
-    def serving_fn(x):
-        return model(x, training=False)
-
-    concrete_fn = serving_fn.get_concrete_function(tf.TensorSpec((1, 128, 3, 1), tf.float32))
-
-    return tf.lite.TFLiteConverter.from_concrete_functions([concrete_fn], model).convert()
+    return tf.lite.TFLiteConverter.from_keras_model(model).convert()
 
 
 def sample_tvm_compile(