Upload Model_Class.py

Model_Class.py

@@ -79,12 +79,13 @@ def load_and_classify_image(image_path, device):
 
 def make_GradCAM(image, device):
 
+    arr = image.numpy().squeeze()
     gpu_model = model.to(device)
     image = image.to(device)
     model.eval()
     target_layers = [gpu_model.model.layer4[-1]]
 
-    arr = image.numpy().squeeze()
+    
     cam = GradCAM(model=gpu_model, target_layers=target_layers)
     targets = None
     grayscale_cam = cam(