Title Update

visualize.py

@@ -60,7 +60,7 @@ def display_cifar_misclassified_data(data: list,
         img = np.transpose(img, (1, 2, 0))
         img = inv_transform(image=img)['image']
         plt.imshow(img)
-        plt.title(r"Treu: " + CLASS_NAMES[data[i][1].item()] + 'Pred: ' + CLASS_NAMES[data[i][2].item()])
+        plt.title(r'Pred: ' + CLASS_NAMES[data[i][2].item()])
         plt.xticks([])
         plt.yticks([])