Fix: Runtime error

app.py

@@ -87,7 +87,7 @@ def inference(input_img,
     target_layers = [eval(f'model.{_layers[target_layer_number-1]}[0]')]
 
     # Get the class activations from the selected layer
-    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, :]