Update compAnIonv1.py

compAnIonv1.py

@@ -76,25 +76,6 @@ def create_bert_classification_model(bert_model,
     input_ids = tf.keras.layers.Input(shape=(MAX_SEQUENCE_LENGTH,), dtype=tf.int64, name='input_ids')
     token_type_ids = tf.keras.layers.Input(shape=(MAX_SEQUENCE_LENGTH,), dtype=tf.int64, name='token_type_ids')
     attention_mask = tf.keras.layers.Input(shape=(MAX_SEQUENCE_LENGTH,), dtype=tf.int64, name='attention_mask')
-
-    class CustomLayer(tf.keras.layers.Layer):
-        def call(self, inputs):
-            if isinstance(inputs, tf.Tensor):
-                # If the input is a tensor, convert it to int64
-                return tf.constant(inputs, dtype=tf.int64)
-            elif isinstance(inputs, dict):
-                # If the input is a dictionary, process each value
-                return {key: tf.constant(value, dtype=tf.int64) for key, value in inputs.items()}
-            else:
-                # If the input type is unknown, raise an error
-                raise ValueError(f"Unsupported input type: {type(inputs)}")
-
-
-    # Create instances of the custom layer for each input
-    input_ids_layer = CustomLayer()(input_ids)
-    token_type_ids_layer = CustomLayer()(token_type_ids)
-    attention_mask_layer = CustomLayer()(attention_mask)
-
                                     
     bert_inputs = {'input_ids': input_ids,
                    'token_type_ids': token_type_ids,