updated app

app.py

@@ -29,6 +29,32 @@ from skimage.color import gray2rgb
 from matplotlib import pyplot as plt
 from torchvision import transforms
 
+from captum.attr import (
+    GradientShap,
+    DeepLift,
+    DeepLiftShap,
+    IntegratedGradients,
+    LayerConductance,
+    NeuronConductance,
+    NoiseTunnel,
+    Saliency,
+    InputXGradient,
+    GuidedBackprop,
+    Deconvolution,
+    GuidedGradCam,
+    FeatureAblation,
+    ShapleyValueSampling,
+    Lime,
+    KernelShap
+)
+
+from captum.metrics import (
+    infidelity,
+    sensitivity_max
+)
+
+from captum.attr._utils.visualization import visualize_image_attr
+
 device = torch.device('cpu')
 opt = get_args(is_train=False)