Update compAnIonv1.py

compAnIonv1.py

@@ -76,7 +76,11 @@ 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')
-    
+    input_ids = tf.convert_to_tensor(input_ids)
+    token_type_ids = tf.convert_to_tensor(token_type_ids)
+    attention_mask = tf.convert_to_tensor(attention_mask)
+
+                                    
     bert_inputs = {'input_ids': input_ids,
                    'token_type_ids': token_type_ids,
                    'attention_mask': attention_mask}