small fixes

app.py

@@ -12,15 +12,12 @@ import torch
 import numpy as np
 
 class Explainer:
-    def __init__(self, model):
-        self.model = model
-        self.default_cmap = LinearSegmentedColormap.from_list('custom blue', 
-                                                    [(0, '#ffffff'),
-                                                    (0.25, '#000000'),
-                                                    (1, '#000000')], N=256)
-
     def __init__(self, model, img, class_names):
         self.model = model
+        self.default_cmap = LinearSegmentedColormap.from_list('custom blue', 
+                                                [(0, '#ffffff'),
+                                                (0.25, '#000000'),
+                                                (1, '#000000')], N=256)
         self.class_names = class_names
 
         transform = transforms.Compose([