small fixes

app.py

@@ -41,7 +41,7 @@ class Explainer:
         with torch.no_grad():
             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)}
 
         self.pred_score, self.pred_label_idx = torch.topk(self.output, 1)