Update load_album.py

load_album.py

@@ -9,7 +9,6 @@
 from pathlib import Path
 from PIL import Image
 from torch.utils.data import Dataset
-from torchvision import transforms as T
 
 
 class AlbumDataset(Dataset):
@@ -36,10 +35,5 @@ class AlbumDataset(Dataset):
         return len(self.imgs)
 
     def __getitem__(self, idx):
-        image_path = self.imgs[idx]
-        image = Image.open(image_path).convert("RGB")  # Ensure 3-channel RGB
-        
-        if self.transform:
-            image = self.transform(image)
-        
-        return image, str(image_path)  # Optionally return the path with the image
+        # not needed
+        pass