Hugging Face's logo

Spaces:
K00B404
/
pix2pix_flux_train
Running

App Files Community
K00B404 commited on
Commit
31c8ff1
·
verified ·
1 Parent(s): 9dda4c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -31,6 +31,7 @@ class Pix2PixDataset(torch.utils.data.Dataset):
31
  # Filter dataset for 'original' and 'target' images
32
  #https://huggingface.co/datasets/K00B404/pix2pix_flux_set/viewer/default/train?f[label][value]=0
33
  #https://huggingface.co/datasets/K00B404/pix2pix_flux_set/viewer/default/train?f[label][value]=1
 
34
  self.originals = [x for x in ds["train"] if x['label'] == 0]
35
  self.targets = [x for x in ds["train"] if x['label'] == 1]
36
 
 
31
  # Filter dataset for 'original' and 'target' images
32
  #https://huggingface.co/datasets/K00B404/pix2pix_flux_set/viewer/default/train?f[label][value]=0
33
  #https://huggingface.co/datasets/K00B404/pix2pix_flux_set/viewer/default/train?f[label][value]=1
34
+ print(ds)
35
  self.originals = [x for x in ds["train"] if x['label'] == 0]
36
  self.targets = [x for x in ds["train"] if x['label'] == 1]
37