Update app.py

app.py

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