Update load_album.py

load_album.py

@@ -28,6 +28,8 @@ class AlbumDataset(Dataset):
         self.imgs = [p for p in self.root_dir.rglob('*') if p.suffix.lower() in ['.jpg', '.jpeg', '.png', '.bmp', '.tiff']]
         if not self.imgs:
             raise ValueError(f"No images found under {root_dir}.")
+
+        self.imgs = sorted(self.imgs)
         
         self.transform = transform