Thanks for clearing my doubt. It is working now.

I am getting an error when tried with 2 neurons and softmax activation func with binary crossentropy loss.

ValueError: logits and labels must have the same shape ((None, 2) vs (None, 1))

what about loss function I need to use?

yes, it is imbalanced