Update app.py

app.py

@@ -70,7 +70,7 @@ def compute_similarity(output1, output2):
 # Function to visualize feature heatmaps
 def visualize_heatmap(model, image):
     model.eval()
-    x = image.unsqueeze(0) # remove batch dimension
+    x = image#.unsqueeze(0) # remove batch dimension
     features = model.convnet(x) # feature heatmap learnt by model
     heatmap = torch.mean(features, dim=1).squeeze().detach().numpy() # normalize heatmap to ndarray
     plt.imshow(heatmap, cmap="hot") # display heatmap as plot