Update app.py

app.py

@@ -31,7 +31,7 @@ def make_gradcam_heatmap(img_array, model, last_conv_layer_name):
     )
     with tf.GradientTape() as tape:
         last_conv_layer_output, preds = grad_model([img_array])
-        class_channel = preds[0][:, 0]
+        class_channel = preds[0]
     grads = tape.gradient(class_channel, last_conv_layer_output)
     pooled_grads = tf.reduce_mean(grads, axis=(0, 1, 2))
     last_conv_layer_output = last_conv_layer_output[0]