Update compAnIonv1.py

compAnIonv1.py

@@ -78,16 +78,16 @@ def create_bert_classification_model(bert_model,
     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)}")
+        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