Update gradcam.py

gradcam.py

@@ -71,7 +71,7 @@ class GradCam():
             return np.hstack(results)
         
         
-    def print_top_categories(self, model, img_tensor, top_k=5):
+    def get_top_category(self, model, img_tensor, top_k=5):
         logits = model(img_tensor.unsqueeze(0)).logits
         probabilities = torch.nn.functional.softmax(logits, dim=1)
         topIdx = logits.cpu()[0, :].detach().numpy().argsort()[-1]