@tf.function(
    input_signature=[tf.TensorSpec((None, 4), tf.float32, name="features")]
)
def classify(features):
    scores = model(features, training=False)
    labels = tf.cast(scores >= 0.5, tf.int32)
    return {"scores": scores, "labels": labels}