Update app.py

app.py

@@ -31,7 +31,10 @@ class Pix2PixDataset(torch.utils.data.Dataset):
         # Filter dataset for 'original' and 'target' images
         self.originals = [x for x in ds["train"] if x['label'] == 'original']
         self.targets = [x for x in ds["train"] if x['label'] == 'target']
-
+        print(f"nr images{len(ds["train"])}")
+        print(f"nr 170{ds["train"][170]}")
+        print(f"nr 171{ds["train"][171]}")
+        
         # Ensure the number of original and target images match
         assert len(self.originals) == len(self.targets), "Mismatch in number of original and target images."