small fixes

app.py

@@ -36,10 +36,10 @@ class Explainer:
 
         self.transformed_img = transform(img)
         self.input = transform_normalize(self.transformed_img)
-        self.input = input.unsqueeze(0)
+        self.input = self.input.unsqueeze(0)
 
         with torch.no_grad():
-            self.output = self.model(input)
+            self.output = self.model(self.input)
             self.output = F.softmax(self.output, dim=1)
         print(self.output.shape)
         self.confidences = {class_names[i]: float(self.output[0, i]) for i in range(3)}