Added files

app.py

@@ -97,7 +97,7 @@ def generate_prediction(input_image, num_classes=3, show_gradcam=True, transpare
     if show_gradcam:
         # Get the target layer
         target_layers = get_target_layer(layer_name)
-        cam = GradCAM(model=model, target_layers=target_layers, use_cuda=False)
+        cam = GradCAM(model=model, target_layers=target_layers)
         cam_generated = cam(input_tensor=input_image, targets=None)
         cam_generated = cam_generated[0, :]
         display_image = show_cam_on_image(orginal_img / 255, cam_generated, use_rgb=True, image_weight=transparency)