Upload Model_Class.py

Model_Class.py

@@ -82,7 +82,7 @@ def make_GradCAM(image, device):
     gpu_model = model.to(device)
     image = image.to(device)
     model.eval()
-    target_layers = [gpu_model.gpu_model.layer4[-1]]
+    target_layers = [gpu_model.model.layer4[-1]]
 
     arr = image.numpy().squeeze()
     cam = GradCAM(model=gpu_model, target_layers=target_layers)