Update app.py

app.py

@@ -42,7 +42,7 @@ def inference(input_img, input_slider_grad_or_not, transparency = 0.5, target_la
         target_layers = [model.layer_2[-1]]
     if target_layer_number == 3:
         target_layers = [model.layer_3[-1]]
-    cam = GradCAM(model=model, target_layers=target_layers, use_cuda=False)
+    cam = GradCAM(model=model, target_layers=target_layers)
     grayscale_cam = cam(input_tensor=input_img, targets=None)
     grayscale_cam = grayscale_cam[0, :]
     visualization = show_cam_on_image(orginal_img/255, grayscale_cam, use_rgb=True, image_weight=transparency)