Update app.py

app.py

@@ -41,7 +41,8 @@ class ColorizationDataset(Dataset):
         img = Image.open(self.paths[idx]).convert("RGB")
         img = self.transforms(img)
         img = np.array(img)
-        img_lab = rgb2lab(img).astype("float32") # Converting RGB to L*a*b
+        try:img_lab = rgb2lab(img).astype("float32") # Converting RGB to L*a*b
+        except:print(img)
         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