Spaces:
Sleeping
Sleeping
Update preprocessing.py
Browse files- preprocessing.py +1 -1
preprocessing.py
CHANGED
@@ -15,7 +15,7 @@ class Text2ImageDataset(Dataset):
|
|
15 |
self.dataset_dir = dataset_dir
|
16 |
with open(os.path.join(self.dataset_dir, 'descriptions.json'), 'r') as file:
|
17 |
self.dataset = json.load(file)
|
18 |
-
self.images_path = os.path.join(dataset_dir
|
19 |
|
20 |
def __len__(self):
|
21 |
return len(self.dataset)
|
|
|
15 |
self.dataset_dir = dataset_dir
|
16 |
with open(os.path.join(self.dataset_dir, 'descriptions.json'), 'r') as file:
|
17 |
self.dataset = json.load(file)
|
18 |
+
self.images_path = os.path.join(dataset_dir)
|
19 |
|
20 |
def __len__(self):
|
21 |
return len(self.dataset)
|