Update app.py

app.py

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