Update app.py

app.py

@@ -38,7 +38,7 @@ def inference(input_img, transparency = 0.5, target_layer_number = -1):
     confidences = {classes[i]: float(o[i]) for i in range(10)}
     _, prediction = torch.max(outputs, 1)
     target_layers = [pytorch_model.layer2[target_layer_number]]
-    cam = GradCAM(model=lightning_model, target_layers=target_layers, use_cuda=False)
+    cam = GradCAM(model=lightning_model, target_layers=target_layers)
     grayscale_cam = cam(input_tensor=input_img, targets=None)
     grayscale_cam = grayscale_cam[0, :]
     img = input_img.squeeze(0)