Update app.py

app.py

@@ -337,10 +337,11 @@ class MyDataset(torch.utils.data.Dataset):
         return len(self.img_list)
 
     def __getitem__(self, idx):
-        img = self.img_list[idx]
+        img = Image.open(self.img_list[idx]).convert("RGB")
+        # img = self.img_list[idx]
         img = self.augmentations(img)
         img = np.array(img)
-        print(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