input_layer = Input(shape=(tokenizer.max_seq_length,), dtype='int32')

bert_output = bert_model(input_layer)

output_layer = Dense(1, activation='sigmoid')(bert_output['pooled_output'])

model = Model(inputs=input_layer, outputs=output_layer)

model.compile(optimizer=Adam(learning_rate=2e-5), loss='binary_crossentropy', metrics=['accuracy'])

        __
        __