Update app.py

app.py

@@ -83,8 +83,8 @@ def create_heatmap(model, imgs, class_index):
     #removing the extra dimension of value 1
     heatmap = tf.squeeze(heatmap)
     
-    #applying relu activation
-    heatmap = tf.keras.activations.relu(heatmap)
+    # For visualization purpose, we will normalize the heatmap between 0 & 1
+    heatmap = tf.maximum(heatmap, 0) / tf.math.reduce_max(heatmap)
     
     return heatmap, preds.numpy()