Heatmap generator for 2 classes

[ghost]

Hi, I am trying to generate heatmap for X-Ray images with my own Inceptionv4 trained model. However, my problem statement doesn't have 14 classes as chexnet. Can you please tell me what changes do i need to do for binary classification?
In this part of code specifically

heatmap = None
for i in range (0, len(weights)):
    map = output[0,i,:,:]
    if i == 0: heatmap = weights[i] * map
    else: heatmap += weights[i] * map