Update app.py

app.py

@@ -41,8 +41,7 @@ class ColorizationDataset(Dataset):
         img = Image.open(self.paths[idx]).convert("RGB")
         img = self.transforms(img)
         img = np.array(img)
-        try:img_lab = rgb2lab(img).astype("float32") # Converting RGB to L*a*b
-        except:print(img)
+        img_lab = rgb2lab(img).astype("float32") # Converting RGB to L*a*b
         img_lab = transforms.ToTensor()(img_lab)
         L = img_lab[[0], ...] / 50. - 1. # Between -1 and 1
         ab = img_lab[[1, 2], ...] / 110. # Between -1 and 1
@@ -341,6 +340,7 @@ class MyDataset(torch.utils.data.Dataset):
         img = self.img_list[idx]
         img = self.augmentations(img)
         img = np.array(img)
+        print(img)
         img_lab = rgb2lab(img).astype("float32") # Converting RGB to L*a*b
         img_lab = transforms.ToTensor()(img_lab)
         L = img_lab[[0], ...] / 50. - 1. # Between -1 and 1