Update app.py

app.py

@@ -54,10 +54,10 @@ def create_heatmap(model, imgs, class_index):
     
     #predicting the images and getting the conv outputs and predictions from the gradcam model
     with tf.GradientTape() as tape:
-    maps, preds = model(imgs);
+        maps, preds = model(imgs);
     
-    # class_channel = tf.expand_dims(preds[:,class_index],axis = 1)
-    class_channel = preds[:, class_index]
+        # class_channel = tf.expand_dims(preds[:,class_index],axis = 1)
+        class_channel = preds[:, class_index]
     
     #computing gradients of predictions w.r.t the feature maps
     if class_index == -1: